在此之前默認Anaconda,CUDA, cuDNN 已安裝好了。CUDA, cuDNN 下載對應版本安裝。
我的環境爲:win10+Anaconda+python3.6+cuda10.0
1、打開anaconda prompt
命令行輸入:conda create --name pytorch_gpu python=3.6
python_gpu爲anaconda下虛擬環境名稱,可自定義,python=3.6爲選擇安裝的python版本。
proceed選擇y,回車, 等待相關包下載,可以看到在安裝完成之後,信息提示:
如果要啓用創建的環境,輸入指令
conda activate pytorch_gpu
關閉該環境的話,輸入指令
conda deactivate
2、安裝pytorch
在剛剛創建的環境中安裝pytorch, 在pytorch 官網 找到對應的下載指令進行下載,選擇與自己環境匹配的,如下是我的環境配置:
在已經激活的環境中輸入上圖紅框中的命令:
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
默認pytorch官網爲下載源,下載速度太慢,很容易報錯,所以更改爲清華大學鏡像,
命令行輸入下面的命令:
# 添加Anaconda的清華鏡像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 設置搜索時顯示通道地址
config --set show_channel_urls yes
然後在輸入:
conda install pytorch torchvision cudatoolkit=10.0
注意要把後面的-c pytorch去掉,不然還是使用的默認源下載。
等待下載和安裝:
3、測試
不報錯,說明安裝成功!!
注:如果要刪除自定義源更換回conda的默認源,直接刪除channels即可,命令如下:
conda config --remove-key channels