????????本文主要記錄如何在電腦中安裝pytorch以及在vscode中配置相關(guān)環(huán)境,以及在配置過程中遇到的一些問題。廢話不多說,現(xiàn)在開始。
一、安裝VS Code
? ? ? ? 前往微軟官網(wǎng)下載vs code,選擇社區(qū)版,默認(rèn)安裝即可
?二、安裝Anaconda
? ? ? ? 前往Anaconda官網(wǎng)下載一個安裝包,要注意自己電腦的Python版本,可以在控制臺中使用
python -v
????????????????查看自己的版本號。我的是3.9所以直接下載最新的
?三、下載CUDA工具
? ? ? ? 1.確定自己的顯卡所支持的CUDA版本
? ? ? ? ? ? ? ? 在英偉達(dá)控制面板的左下角找到“系統(tǒng)信息”,在“組件”中找到支持的CUDA版本
? ? ? ? ? ? ? ? ?前往CUDA下載頁面,選擇自己電腦合適的CUDA版本進(jìn)行下載。如果在默認(rèn)頁面找不到對應(yīng)的版本,可以在下面查找歷史版本。
? ? ? ? ? ? ? ? ?安裝時可以選擇自定義安裝,以避免安裝全家桶(其實早都裝完了),我選擇勾掉VS Integration,其他的全部安裝。當(dāng)然,如果C盤內(nèi)存不夠了也可以指定其他盤,并沒有什么影響。
? ? ? ? ?安裝完成后一般會自動部署環(huán)境,如果沒有自動部署環(huán)境的話可以百度查看一下如何部署
? ? ? ? 隨后在CMD中使用代碼驗證環(huán)境
nvcc --version
? ? ? ? ?2.下載CUDnn模塊
? ? ? ? ? ? ? ? CUDnn用于神經(jīng)網(wǎng)絡(luò)計算,下載地址為CUDnn下載,完成注冊后選擇自己對應(yīng)的CUDA版本進(jìn)行下載,下載之后,將壓縮包中的三個文件夾解壓到之前安裝CUDA的地方
? ? ? ? ? ? ? ? ?具體可以參見這篇博文CUDA安裝教程
三、安裝PyTorch
? ? ? ? 前往官網(wǎng)https://pytorch.org/,選擇自己對應(yīng)的情況,獲得一句安裝口令
? ? ? ? ?在Anaconda的控制臺中直接粘貼這句話,等待安裝成功即可。如果出現(xiàn)done則表示安裝成功,國內(nèi)部分網(wǎng)絡(luò)可能出現(xiàn)下載過慢/安裝失敗的情況,可以參考如何解決pytorch下載慢的問題,更換清華鏡像來解決。
四、配置VSCode
? ? ? ? 1.安裝插件
????????????????建議安裝以下插件
? ? ? ? ?2.配置解釋器
? ? ? ? ? ? ? ? 在vscode的右下角選擇環(huán)境
? ? ? ? ? ? ? ? ?選擇pytorch解釋器,沒有的話可以到安裝路徑中查找
? ? ? ? ? ? ? ? ?如果運行出錯的話可以嘗試配置一下環(huán)境(反正我是沒配置,有需要的話可以百度搜索)
? ? ? ? 3.一點小問題
? ? ? ? ? ? ? ? 我在完成配置后進(jìn)行測試發(fā)現(xiàn)numpy的版本號與pytorch版本號不一致,可以使用
pip list
or
conda list
? ? ? ? ? ? ? ? ? ? ? ? 來核驗版本號,如果不對的話記住編譯器提示你的正確版本號,在Anaconda的控制臺中運行以下代碼即可解決
pip uninstall numpy
pip install numpy==1.18.2 --替換為你需要的版本
最后完美運行文章來源:http://www.zghlxwxcb.cn/news/detail-835030.html
文章來源地址http://www.zghlxwxcb.cn/news/detail-835030.html
到了這里,關(guān)于神經(jīng)網(wǎng)絡(luò)(十五)在VS Code下搭建PyTorch環(huán)境的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!