首先要确定你torch包是能用的,不报错
第二,
去conda清华源找到对应的pytorch gpu版本(cuda版本)
打开 Index of /anaconda/cloud/pytorch/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
比如你下的是python3.8,cuda11.1,你就去找那个包
直接ctrl+f 搜文件名即可
例如 pytorch-1.11.0-py3.7_cuda11.1_cudnn8_0.tar.bz2
之后的话找到你的虚拟环境,进入之后直接找到下载文件根目录
输入
conda install --offline pytorch-1.11.0-py3.7_cuda11.1_cudnn8_0.tar.bz2
之后pycharm检查会发现已成功,这个问题是因为你之前下载文件的时候他会自动给你推荐cpu的torch包,gpu根本没装上
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3