Pytorch安裝踩坑之路

我擦，Pytorch環境的搭建不得不說，真的有點煩，可能是自己菜吧。

重點：不要像我一樣直接用cuda10.0的剛1.5版本的pytorch，你嘗試過就知道，各種包衝突，然後在Anaconda Prompt上提示find shortest time to load xxx(這無非是一種尋找響應時間最短的一種算法，再根據結果去拉相應的包)，沒用還會衝突。

1、選對正確的cuda版本

       首先，先確保如果想下載pytorch1.5版本或以上的並且想要使用GPU加速的，確保CUDA版本在10.1及以上，10.0官網已經不支持了。如下圖：

       現在只需要安裝對應版本的cuda就行了，安裝pytorch的時候會自動下載適配的cudnn。

2、更改鏡像

       在Anaconda Prompt命令行中執行下面的指令。

	# 配置國內源方便安裝Numpy,Matplotlib等
	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/
	# 配置國內源安裝PyTorch用
	conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
	# 顯示源地址
	conda config --set show_channel_urls yes

       執行完上面的指令後，可以查看到下面文件中相應內容的變化。

3、創建單獨的pytorch虛擬環境

       在Anaconda Prompt命令行中運行conda create -n pytorch python=3.7.3。這裏選擇自己的python版本。

       執行之後發現要強制更新anaconda，沒事問題不大，按照提示執行對應的指令即可。

       運行之後，又報錯了。其實是包的兼容性問題。那問題不大，找到那個對應的包名，直接移除就好了。

4、安裝pytorch

       那此時再次創建虛擬環境已經可以成功了。使用conda activate pytorch切換到新的虛擬環境中。

       要是中間斷了，繼續執行該指令，會將未下完的包重新下載。

5、重新安裝anaconda-navigator

       執行上面的conda remove pillow之後，anaconda-navigator打不開了。需要切換到base的虛擬環境再執行下面的指令重新安裝。

	conda install anaconda-navigator

       然後anaconda-navigator就可以正常打開了。

6、測試pytorch環境

       進入自建的pytorch虛擬環境。執行下面的python代碼。

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

       看到下面的結果說明安裝成功了。

7、整合PyCharm

       OK，整合完畢。