終於在和環境的鬥爭中勝利了。
之前安裝不上gpu版本大概是CUDA的原因,沒有搞懂CUDA到底要裝哪個版本,最主要的是到底要不要裝。
大概碰到了以下幾個問題。
- 顯卡驅動是否安裝
學長的判斷方法,在桌面右鍵,看有沒有NVIDIA控制面板。如果有,說明已經安裝好了顯卡驅動。沒有的話,就需要去官網下載驅動安裝了。
- 下載Anaconda
利用Anaconda安裝python和pytorch等包,是非常方便的。
下載Anaconda可以到官網,建議到清華的鏡像網站 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 清華鏡像
直接拉到最下面,下載最新的就好啦。
最下面的是python3.7版本的Anaconda.
- 安裝pytorch-gpu
以往在這裏踩坑最多。
打開Anaconda的Anaconda Prompt。(從開始那裏就能找到)
首先,爲了下載得快,或者下載不被中斷,建議添加清華的鏡像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
接着,需要知道要不要下載CUDA!!!有些電腦是已經安裝好CUDA的。
查看自己的CUDA信息:桌面空白處右鍵--->NVIDIA控制面板--->左下角有系統信息---->點擊“組件”--->查看NVIDIA CUDA.DLL
於是我發現自己根本不需要下載安裝CUDA23333,因爲已經安裝好了CUDA 10.1版本。
最後,只需要輸入對應的10.1版本的Pytorch-gpu版本的命令:
conda install pytorch torchvision cudatoolkit=10.1
會下載很多包,選Y就完事了。
環境應該是裝在base下的,沒有自行創建環境。
- 驗證是否安裝成功
命令行輸入python,敲以下的代碼
import torch
import torchvision
print(torch.cuda.is_available())
返回爲True則安裝成功。