如何在 Linux 服务器上安装 pytorch

一.查看服务器的cuda版本

使用如下命令：

nvcc -V

博主的cuda版本为10.1

二.安装对应版本的pytorch

进入pytorch官网 https://pytorch.org/ 选择对应的版本

输入提示的安装命令，安装过慢可以使用镜像源

pip install torch torchvision -i  https://pypi.tuna.tsinghua.edu.cn/simple

三.检查是否安装成功

>>> import torch
>>> torch.__version__

显示pytorch版本为1.3.0，安装成功！