淘先锋技术网

首页 1 2 3 4 5 6 7

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