Linux18.04 安装pytorch 和torchvision
cuda 安装请看上一篇博客.Linux18.04 cuda环境配置
pytorch官网
1. pytorch安装
本文从最简单最快的方式安装,安装之前先看看官网torch对应的torchvision版本.
开始安装
whl文件下载地址
1.从whl文件下载地址中复制torch和torchvision的地址,然后执行.
pip install https://download.pytorch.org/whl/cu111/torch-1.8.1%2Bcu111-cp36-cp36m-linux_x86_64.whl
2. torchvision安装
https://download.pytorch.org/whl/cu111/torchvision-0.9.1%2Bcu111-cp36-cp36m-linux_x86_64.whl