在windows下用anaconda安裝 pytorch

概述

本次主要是談論下如何在windows10下面安裝pytorch的gpu版本，第一次寫博客，可能寫的有點囉嗦，還望大家耐心看下去。

1.安裝anaconda

這個網上的教學很多，百度一下即可，要注意的是由於外網不科學上網速度賊慢，需要添加清華園，在命令行（win+r再輸入cmd）或者anaconda prompt中輸入如下配置。如果已經裝過anaconda但是在安裝時出現環境之類的錯誤時，建議卸掉重新安裝，我當時出現了這種問題重新安裝便解決了。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

2.查看自己的顯卡驅動版本（很重要！！！）

之前大概是因爲版本不匹配，我安裝的gpu版本的pytorch，但是gpu無效！打開控制面板，點擊卸載程序，查看NVIDIA的版本，下圖中可以看出我的版本是441.66。

然後打開 這裏 在網頁中下拉，查看錶格，找到適合自己型號的CUDA版本，由於我是重新安裝的最新版本（2019年1月26日），且是windows10環境，版本爲441.66，滿足最右列的所有值，所有版本的CUDA我均可以安裝，但是注意，anaconda會自動幫我們安裝CUDA Toolkit，我們不需要安裝CUDA和CUDNN！這裏需要記住能用哪些版本的CUDA，後面要用到！

可能遇到的問題

你可能會發現你的顯卡驅動，版本很低，並且低版本可能會在後續的安裝中出現很多很多的問題，比如版本不匹配，國內源不支持等等問題。如果出現了，建議卸載顯卡驅動，安裝較新版本的驅動。這裏百度即可，需要注意的是卸載後需要重啓電腦，否則會無法安裝。下面主要談談一些重要的方面。

顯卡型號

右擊此電腦，點擊管理，點擊設備管理器，點擊顯示配置器，下圖可以看到，我的顯卡型號是NVIDIA GeForce GTX 1050

點擊 這裏 ，安裝剛剛找到的顯卡型號，搜索你所需要的驅動型號，點擊search。這裏Download Type的Studio Driver是時候我們學習辦公使用的，而Game Ready Driver則是適合遊戲的，這裏我選擇的Studio Driver。

這是我搜索到的結果，點擊下載即可，然後就按照網上我教學安裝吧。

3.安裝pytorch

登錄官網 這裏 ，下拉網頁，按照電腦的配置選擇版本。我這裏是選擇的stable穩定版本，windows，用conda安裝，py是3.7版本，CUDA前面查到了是10.2以下都可以用，最後得到了cmd下的運行語句 conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

需要注意的是最後運行的時候要去掉最後的-c pytorch，這代表從官網下載，而不是國內源。
所以運行下面語句

conda install pytorch torchvision cudatoolkit=10.1

注意點

一定要去掉最後的-c pytorch，當然如果你能fq另當別論。此外，需要注意版本的選取，建議選取運行語句最後只有-c pytorch的，而沒有別的內容的，比如下圖的內容多出了-c defaults和-c numba/label/dev，這裏當時對我造成了很大的困擾，因爲國內源沒有某些資源，下載速度慢的讓人抓狂，並且最後成功安裝後gpu無效，可能是我當初沒有重裝顯卡驅動的緣故。總之這樣之後最後的安裝時間不到二十分鐘。

判斷是否安裝成功

運行如下代碼，返回True則表示gpu版本運行成功，False則是gpu無法使用，但是torch照常能用，就會後面操作可能會很慢。

import torch
print(torch.cuda.is_available())

運行成功！

總結

第一次寫博客，還望多多包涵。