Pytorch安裝教程（Linux)

安裝准備

1. 請先安裝Anaconda或者Miniconda並配置好清華源
2. 輸入命令nvidia-smi，查看CUDA版本，可以看到是10.1
   

安裝教程

1. 用conda新建一個名為pytroch的環境

conda create -n pytorch python=3.7

2. 激活剛才創建的名為pytorch的環境

conda activate pytorch

3. 進入Pytorch的官網，CUDA版本選擇之前我們查詢的。然后復制下方的安裝命令
   
4. 在第2步激活的pytorch環境中運行第3步復制的安裝命令

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

驗證安裝

1. 驗證torch能不能正常使用，正常打印則說明安裝成功

import torch
x = torch.rand(5, 3)
print(x)

2. 再來看看CUDA是否能用，如果打印True則說明能用

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

引用參考

https://pytorch.org/get-started/locally/