一、使用anaconda安装
- 安装命令查看官网,根据cuda版本等要求选择
- cuda10.1对应的最低版本是1.4.0,安装命令
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch
- cuda10.0对应的最高版本是1.2.0,安装命令
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
- cuda9.0对应的最高版本是1.1.0,安装命令
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
二、pip安装
- 在这里寻找安装链接
- 也可以把whl文件下载下来安装