问题:torch\lib\caffe2_nvrtc.dll" or one of its dependencies
环境:
系统:win10
环境:conda 22.9.0 cuda version: 11.6
问题分析:
按pytorch官网提示命令来anaconda安装pytorch后,测试import torch x = torch.rand(5, 3)
print(x)出错。错误提示找不到torch\lib\caffe2_nvrtc.dll" or one of its dependencies。查看对应目录下已经存在了这个caffe2_nvrtc.dll,猜测是cuda支持问题。
解决步骤:
一、命令行执行nvidia-smi查看cuda版本,下载对应版本cuda,下载网址:https://developer.nvidia.com/cuda-toolkit-archive,下图所示是本机版本对应的
二、安装cuda。
1、安装下载的cuda套件。
2、命令行执行nvcc --version 查看版本,如果显示正常说明cuda已经安装完成。
三、再次测试torch,发现已经可以执行成功。文章来源:https://www.toymoban.com/news/detail-531024.html
Tips: 只是提供解决问题的一个参考,如有类似问题或者cuda版本不对的,可以更换版本以后测试。文章来源地址https://www.toymoban.com/news/detail-531024.html
到了这里,关于解决:torch\lib\caffe2_nvrtc.dll“ or one of its dependencies的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!