admin管理员组

文章数量:1626244

​
​

活动地址：CSDN21天学习挑战赛

1.gpu相关

  • 可以参考这里

2.pytorch

  • 本来的cuda是11.2，但并没有与之对应的pytorch版本，选择下载cuda11.3对应的版本，网上的资料都说没有发现问题
  • 官网，参考官网得到下载的命令：conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

3.换源

  • 国内清华镜像源地址，在里面找自己电脑的系统，将下面的地址做出响应的修改
  • conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu/anaconda/cloud/pytorch/win-64/

4.验证

(base) PS C:\Users\53423> python
Python 3.9.12 (main, Apr  4 2022, 05:22:27) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>>

5.常见错误

  • 清华源常常会下载cpu版本而不是gpu版本
  • 解决办法：指定具体版本（参考清华源中都有什么版本）
  • 例如：
pip install pytorch==1.9.1-py3.7_cuda10.2_cudnn7_0 torchvision torchaudio cudatoolkit==10.2 -f https://mirrors.tuna.tsinghua.edu/anaconda/cloud/pytorch/win-64/
  • 或者在清华源手动下载，离线安装，参考这里，效果如下：

本文标签： 清华镜像版本国内GPU

版权声明：本文标题：Windows10、anaconda3、python3.9、国内清华镜像源安装gpu版本pytorch 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://m.elefans.com/xitong/1728942959a1180940.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。