一.查看服务器的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,安装成功!