Windows 10系統在Anaconda下安裝GPU版Pytorch

在此之前默認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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章