一、使用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文件下载下来安装
本文提供Anaconda和pip两种方式安装PyTorch的详细指令,针对不同CUDA版本选择合适的PyTorch、Torchvision和cudatoolkit。例如,CUDA 10.1对应PyTorch 1.4.0,CUDA 10.0对应1.2.0,CUDA 9.0对应1.1.0。确保正确匹配版本以实现高效GPU加速。
7862

被折叠的 条评论
为什么被折叠?



