1.使用Pip更新Pytorch和torchvision
# 更新pytorch和torchvision安装包
pip3 install --upgrade pytorch torchvision
2.使用pip安装指定版本
3. 查看linux下pytorch的版本
import torch
print(torch.__version__) #注意是双下划线
4.查看 CUDA 版本
cat /usr/local/cuda/version.txt
5.查看 CUDNN 版本
6.安装下载好的python库安装包whl
- 在pytorch官网下载whl安装包
torch-1.5.0-cp36-cp36m-manylinux1_x86_64.whl - linux系统下cd到虚拟编译环境或者直接在终端中使用命令:
pip3 install torch-1.5.0-cp36-cp36m-manylinux1_x86_64.whl
7.tar或者tar.gz文件解压
tar -xvf file.tar //解压 tar包
tar -xzvf file.tar.gz //解压tar.gz