活动地址: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/
- 或者在清华源手动下载,离线安装,参考这里,效果如下:
更多推荐
Windows10、anaconda3、python3.9、国内清华镜像源安装gpu版本pytorch
发布评论