如果不能科学上网,使用镜像安装pytorch大概率默认cpu版。直接使用清华源下载离线安装。
我的显卡是3060ti台式,我使用以下版本:
pytorch-2.0.1-py3.10_cuda11.8_cudnn8_0.tar.bz2
torchaudio-2.0.2-py310_cu118.tar.bz2
torchvision-0.15.2-py310_cu118.tar.bz2
清华源网址:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
直接找到对应版本下载。
对于cudatoolkit直接如下安装就行
conda install cudatoolkit=11.8 -c 源地址
安装完成后使用cmd import torch大概率报错,但用pycharm只是显示缺包,从新pip install torchvision后pycharm正常,但是cmd窗口还是显示OSError: [WinError 126],很奇怪。
我是小白,我也不知道这样能不能解决问题,希望有帮助吧。
下图是cmd,还是报错
下面是pycharm,正常