项目场景:
出现这个错误:RuntimeError: CUDA error: no kernel image is available for execution on the device
问题描述
RuntimeError: CUDA error: no kernel image is available for execution on the device
出现这个错误的原因就是因为版本不匹配的问题
根据版本进行匹配
版本参考链接:https://blog.csdn.net/weixin_42069606/article/details/105198845
解决方案:
这是GPU 的版本号
根据版本号确定CUDA的版本号
因为最高是11.3的版本,所以我就按照这个版本按照
conda install pytorch cudatoolkit=11.3 -c pytorch
就是有点太大了,哈哈