1.先检查自己cuda版本:
java
print(torch.version.cuda) #查看cuda版本
print(torch.cuda.is_available()) # 查看cuda是否可用
print(torch.cuda.device_count()) # 查看可行的cuda数目
如果版本高于11建议先降版本,然后再试下。
2.重新安装nvrtc-builtins64_117.dll
- 下载缺失文件
- 拷贝到指定目录C:\ Windows\ System32
- 点击运行即可
3.文件已上传到资源,文章顶部可见