1.安裝anaconda
官網(wǎng)https://www.anaconda.com 下載安裝,路徑全英文然后記得有一步添加path路徑勾選上
2.安裝CUDA
確認(rèn)CUDA版本
這一步一定要確認(rèn)自己電腦的GPU是否符號(hào)版本要求,打開anaconda prompt,輸入nvidia-smi查看電腦GPU狀態(tài);
CUDA version:12.0代表你的電腦所能安裝的CUDA的最高版本,CUDA版本向下兼容,可以安裝比12.0小的CUDA版本;
若電腦上已經(jīng)安裝CUDA,安裝新的版本之前一定要先卸載舊版本具體方法參考:
卸載CUDA
確認(rèn)CUDA和pytorch版本
根據(jù)官網(wǎng):https://pytorch.org/get-started/previous-versions/
找到自己要安裝的pytorch版本選擇對(duì)應(yīng)的CUDA版本
安裝CUDA
一定要根據(jù)上面兩步確認(rèn)自己的電腦應(yīng)該安裝哪個(gè)版本的CUDA避免后續(xù)出現(xiàn)問題;
官網(wǎng)下載:https://developer.nvidia.com/cuda-toolkit-archive
下載成功后雙擊.exe打開進(jìn)行安裝,有一步不選擇精簡,選擇自定義安裝。其他安裝按照默認(rèn)步驟進(jìn)行即可,記住自己的安裝路徑;
3.安裝cudnn
官網(wǎng)下載cudnn
一定要選擇和CUDA版本匹配的版本
注意這里下載需要注冊(cè),可以復(fù)制鏈接和迅雷一起食用效果更佳
下完成后,解壓到本地,打開之前cuda的安裝目錄:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
找到你安裝的版本目錄,打開,找到bin、include、lib目錄,將cuDNN壓縮包內(nèi)對(duì)應(yīng)的文件復(fù)制到bin、include、lib目錄。
4.安裝Pytorch
打開anaconda prompt
pytorch和python對(duì)應(yīng)的版本:https://blog.csdn.net/Y2894297258/article/details/130775874
conda enc list #查看已經(jīng)存在的虛擬環(huán)境
conda remove -n pytorch_c --all #刪除名為pytorch_c的虛擬環(huán)境
conda create -n pytorch_x python=3.6 #創(chuàng)建名為pytorch_x的虛擬環(huán)境,python的版本根據(jù)需要選擇
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch #根據(jù)pytorch官網(wǎng)給的conda命令進(jìn)行安裝,直接官網(wǎng)安裝可能慢一些,可以使用國內(nèi)的鏡像源
安裝完成后,輸入
python
import torch
torch.__version__
torch.cuda.is_avalible() #true則為安裝成功GPU版本的
5.vscode配置
安裝VScode
從官網(wǎng)下載,然后默認(rèn)安裝文章來源:http://www.zghlxwxcb.cn/news/detail-654971.html
vscode配置pytorch環(huán)境
安裝python插件
添加Python解釋器
按快捷鍵 “Ctrl+Shift+P” ,調(diào)出全局設(shè)置搜索窗口,然后輸入 “Python:Select Interpreter” 后會(huì)出現(xiàn) “Python:Select Interpreter” 選項(xiàng),點(diǎn)擊該選項(xiàng);注意這里的python.exe要選擇自己的pytorch環(huán)境下的;
激活pytorch環(huán)境
打開一個(gè)終端
若是PS中斷輸入文章來源地址http://www.zghlxwxcb.cn/news/detail-654971.html
cmd #回到CMD
conda activate pytorch_c #激活pytorch環(huán)境
到了這里,關(guān)于Win10+anaconda+CUDA+pytorch+vscode配置的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!