解决pytorch capability sm_86 is not compatible with the current PyTorch installation

cuda toolkit 版本太低(或者pytorch版本太低)

conda install pytorch==1.7.1 cudatoolkit=11.0 -c pytorch


版权声明:本文为ltochange原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。