linux安裝pytorch

pytorch官網：https://pytorch.org/

一、安裝的整體思路

1.查看自己設備的信息，確定好信息之后去官網選擇自己需要安裝的版本，官網選擇好保本之后會彈出一條命令，將該命令輸入到服務器的中斷就可以正常安裝了

2.下載過程很慢，可以使用國內清華源安裝。

參考：https://www.cnblogs.com/StarZhai/p/15478057.html 　　

 

 二、相關版本的匹配問題

1.查看自己服務器的顯卡和驅動信息

nvidia-smi

 

三、pytorch是否安裝完成的檢查方法。

python

>>> import torch

>>> torch.version
#檢查GPU版本的torch是夠安裝完成
import torch
torch.cuda.is_available

 

 

 

參考：https://www.cnblogs.com/guochaoxxl/p/14646284.html