RTX3090+CUDA11.1 使用Pytorch1.7.1的CUDA问题

解决日期:2021年3月1号

出现报错信息:

no kernel image is available for execution on the device
GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.

解决方案,CUDA和目前的Pytorch不能正常兼容,解决方法和很简单,下载

nightly Pytorch 1.8.1,然后执行版本替换即可。

参阅,不得不说G比B的找错效率要高的多。

你可能感兴趣的:(Pytorch)