1、查看CUDA版本:命令nvcc --version
2、查看python3版本
3、下载相对应版本的Pytorch安装包
安装包下载地址:https://download.pytorch.org/whl/torch_stable.html
cpu对应与cpu版本,cu对于cuda版本
比如我的python版本是3.7,cuda版本是11.1
因此我下载的cuda版本是torch-1.8.1+cu111-cp37-cp37m-linux_x86_64.whl
cp37对应于python3.7,cu111对应于cuda版本11.1
4、安装Pytorch
首先执行:
pip install torchvision
然后进入安装包的路径,接着执行:
pip install torch-1.8.1+cu111-cp37-cp37m-linux_x86_64.whl
5、然后验证pytorch和torchvision是否安装成功,在python命令行输入import torch,无警告代表成功。
autopilot@autopilot-Default-string:~/下载$ python3
Python 3.7.11 (default, Jul 27 2021, 14:32:16)
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>>
至此,Pytorch安装成功!