1、首先可能是自己的python版本、cuda版本、torch和 torchvision 版本是否兼容,我建议是重新安装cuda和torch以及torchvision 的版本。如果你的python版本>3.8的,那么解决方案如下:
(1)首先查看自己的服务器CUDA版本:win+R进入到控制台输入:nvidia-smi 查看点击电脑支持的cuda版本
(2)然后进入https://pytorch.org/get-started/locally/ 挑选符合自己的版本,复制安装命令。
(3)打开Anaconda Prompt控制台,进入到自己要安装的环境下面去,如果你不是使用anaconda去管理python环境的话,你就按照你自己的方式进入到要安装的环境下面去就行,然后执行你刚刚复制的这段命令。
(4)然后再到pycharm中去检测是否cuda可用,我这边是已经ok了的,如果不行,就重新打开一下pycharm
至此可能还会遇到另外一个问题,就是运行后遇到:
ImportError: DLL load failed while importing _imaging: 操作系统无法运行 %1
请参考:https://blog.csdn.net/YCCX_LHH21/article/details/140253553?spm=1001.2014.3001.5501