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

原文地址:https://www.cnblogs.com/StarZhai/p/15459856.html