【簡記】Win10利用anaconda安裝pytorch-gpu

終於在和環境的鬥爭中勝利了。

之前安裝不上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則安裝成功。

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