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,然后执行版本替
原创
2021-03-01 19:31:39 ·
3212 阅读 ·
4 评论