神經網絡學習小記錄48——windows下的torch=1.2.0環境配置
學習前言
好多人問環境怎麼配置,還是出個教程吧。
環境內容
torch:1.2.0
torchvision:0.4.0
Anaconda安裝
最新版本的Anaconda沒有VSCODE,如果大家爲了安裝VSCODE方便可以直接安裝舊版的Anaconda,百度網盤連接如下。也可以裝新版然後分開裝VSCODE。
鏈接: https://pan.baidu.com/s/12tW0Oad_Tqn7jNs8RNkvFA
提取碼: i83n
取網上搜索Anaconda的官網:https://www.anaconda.com/distribution/
下載左邊的python3.7版本,64位的,下載完成後打開:
選擇安裝的位置,可以不安裝在C盤。
我選擇了Add Anaconda to my PATH environment variable,我用起來覺得更好用。
等待安裝完之後,就安裝了Anaconda了。
下載Cudnn和CUDA
我這裏使用的是torch=1.2.0,官方推薦的Cuda版本是10.0,因此會用到cuda10.0,與cuda10.0對應的cudnn是7.4.1.5,這個組合我實驗過了,絕對是可以用的。
cuda10.0官網的地址是:
cuda10.0官網地址
cudnn官網的地址是:需要大家進去後尋找7.4.1.5。
cudnn官網地址
官網下載是比較慢的,可以在百度雲上下載,我給大家分享一手。
鏈接: https://pan.baidu.com/s/1znYSRDtLNFLufAuItOeoyQ
提取碼: 8ggr
下載完之後得到這兩個文件。
下載好之後可以打開cuda_10的exe文件進行安裝。
這裏選擇自定義。
然後直接點下一步就行了。
安裝完後在C盤這個位置可以找到根目錄。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
然後大家把Cudnn的內容進行解壓。
把這裏面的內容直接複製到C盤的根目錄下就可以了。
配置torch環境
Win+R啓動cmd,在命令提示符內輸入以下命令:
創建環境:
conda create –n pytorch python=3.6
激活環境:
activate pytorch
打開pytorch的官方安裝方法:
https://pytorch.org/get-started/previous-versions/
官網推薦的安裝代碼如下,我使用的是Cuda10的版本:
# CUDA 10.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
# CUDA 9.2
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
# CPU only
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
需要注意的是,直接這樣安裝似乎特別慢,因此我們可以進入如下網址:
https://download.pytorch.org/whl/torch_stable.html
找到自己需要的輪子下載。
下載的時候使用迅雷下載就行了,速度還是比較快的!
下載完成後找到安裝路徑:
在cmd定位過來後利用文件全名進行安裝就行了!
這裏我也傳一個百度網盤的版本。
鏈接: https://pan.baidu.com/s/14-QVk7Kb_CVwaVZxVPIgtw
提取碼: rg2e
全部安裝完成之後重啓電腦。
安裝VSCODE
我個人喜歡VSCODE……所以就安裝它啦。
打開anaconda,切換環境。
安裝VSCODE,安裝完就可以launch一下了,之後就可以把VScode固定到任務欄上,方便打開。
最新版本的Anaconda沒有VSCODE因此可以直接百度VSCODE進行安裝。
安裝完成後在左下角更改自己的環境就行了。