本文共 136 字,大约阅读时间需要 1 分钟。
加入了:os.environ["CUDA_VISIBLE_DEVICES"] = '0'
export CUDA_VISIBLE_DEVICES=0
后期又出现:CUDA_ERROR_LAUNCH_FAILED;解决方法如下:
将8.0改成9.0,最后reboot就OK了