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