如何在 Linux 服务器上安装 pytorch

一.查看服务器的cuda版本

使用如下命令:

nvcc -V

博主的cuda版本为10.1

二.安装对应版本的pytorch

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

如何在 Linux 服务器上安装 pytorch_第1张图片

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

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

如何在 Linux 服务器上安装 pytorch_第2张图片

三.检查是否安装成功

>>> import torch
>>> torch.__version__

如何在 Linux 服务器上安装 pytorch_第3张图片

显示pytorch版本为1.3.0,安装成功!

你可能感兴趣的:(软件安装,如何在,Linux,服务器上安装,pytorch,安装,pytorch,在,Linux,服务器上安装,pytorch)