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