win10從零安裝配置pytorch全過程圖文詳解
1.安裝anaconda
(anaconda內(nèi)置python在內(nèi)的許多package,所以不用另外下載python)
可以點擊下面的清華開源軟件鏡像站,在官網(wǎng)下載anaconda不如在這下的快
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 鏡像站
我下載的是下圖這個版本,對應(yīng)的是3.7,anaconda3代表對應(yīng)python3
下載過程中基本上都是默認選項
下面這步可以不勾選add anaconda to my PATH environment variable,事后可以再添加
如果有下面這步,記得跳過,我這個版本的anaconda是沒有這步的
下面檢驗一下安裝成功了沒有,在開始打開anaconda prompt,看到如圖所示即成功
2.確認顯卡及及時更新(為之后的pytorch安裝作準備)
打開任務(wù)管理器-性能,查看有GPU還是只有集顯
再win+r,運行cmd,輸入nvidia-smi后回車
我這里發(fā)生如下報錯,在環(huán)境變量path里面添加C:\Program Files\NVIDIA Corporation\NVSMI 即可解決
正常運行得到的
這里需要注意的是,表中的Driver Version需大于396.26,否則就去geforce experience更新顯卡吧,我這里更新了最新版。
3.下載pytorch
https://pytorch.org/官網(wǎng)
這里需要注意的是,如果之前第二步查看時有GPU,則CUDA選擇10即可,如果選9,之后不好用鏡像源下載,速度會很慢;如果只有集顯,那么CUDA選擇None即可
之后我們復制Run this Command里的安裝指令
打開anaconda prompt,輸入下列指令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/
既然選擇了用鏡像源下載,那么安裝指令的 -c pytorch需刪去,也就是:conda install pytorch torchvision cudatoolkit=10.1
這里我們創(chuàng)建虛擬環(huán)境conda create -n your_env_name(這里就是你給虛擬環(huán)境取的名字,筆者取名pytorch) python=X.X(筆者選下3.6),之后在里面操作,輸入指令conda install pytorch torchvision cudatoolkit=10.1,然后y(es)
最后我們檢驗安裝是否成功,先鍵入python進入python工作環(huán)境,然后鍵入import torch,如果未報錯,則pytorch安裝成功,之后輸入指令torce.cuda.is_available(),如果顯示Turn,則GPU可被pytorch使用
4.pycharm配置
打開pycharm,文件-新項目-Pure Python
Location自行選擇,Interpreter選擇打開你安裝anaconda的位置,選擇envs/your_env_name(虛擬環(huán)境名)/python.exe即可
創(chuàng)建后 在python console
處運行再次運行import torch
,無報錯后運行torch.cuda.is_available()
,如果顯示Turn,則配置
總結(jié)
到此這篇關(guān)于win10從零安裝配置pytorch全過程的文章就介紹到這了,更多相關(guān)window10 pytorch安裝內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Django中對數(shù)據(jù)查詢結(jié)果進行排序的方法
這篇文章主要介紹了Django中對數(shù)據(jù)查詢結(jié)果進行排序的方法,利用Python代碼代替SQL進行一些簡單的操作,需要的朋友可以參考下2015-07-07使用python操作lmdb對數(shù)據(jù)讀取的實例
這篇文章主要介紹了使用python操作lmdb對數(shù)據(jù)讀取的實例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-12-12