我的版本:linux系统
我是在docker中创建的虚拟环境,使用的是python=3.6;
然后出现了 untimeError: CUDA error: no kernel image is available for execution on the deviceCUDA kerne
搜了一圈发现是因为3.6和cuda11.4不兼容的问题;
我重新搞了一个3.7的虚拟环境完美解决!torch目前还没有cuda11.4对应的版本,我就用cuda11.3对应的torch就可以
我的版本:linux系统
我是在docker中创建的虚拟环境,使用的是python=3.6;
然后出现了 untimeError: CUDA error: no kernel image is available for execution on the deviceCUDA kerne
搜了一圈发现是因为3.6和cuda11.4不兼容的问题;
我重新搞了一个3.7的虚拟环境完美解决!torch目前还没有cuda11.4对应的版本,我就用cuda11.3对应的torch就可以