Linux安装pytorch

Linux非root安装Pytorch

  • 1. 查看cuda版本
  • 2. 找到对应cuda版本的pytorch安装命令
  • 3. 在terminal里边粘贴命令,回车安装pytorch

1. 查看cuda版本

输入nvidia-smi查看服务器cuda版本
如下图服务器cuda版本是10.1
Linux安装pytorch_第1张图片

2. 找到对应cuda版本的pytorch安装命令

进入pytorch官网,点击对应的系统、语言、cuda版本等,然后复制命令。
或点击previous version of pytorch查找以前的版本,再复制pytorch安装命令。
Linux安装pytorch_第2张图片
Linux安装pytorch_第3张图片

3. 在terminal里边粘贴命令,回车安装pytorch

在这里插入图片描述
Linux安装pytorch_第4张图片

输入y确认安装以上的包,安装时间可能有点长,请耐心等待。

你可能感兴趣的:(Linux小白,pytorch,linux,深度学习)