
加入了:os.environ["CUDA_VISIBLE_DEVICES"] = '0'

export CUDA_VISIBLE_DEVICES=0

后期又出现:CUDA_ERROR_LAUNCH_FAILED;解决方法如下:

将8.0改成9.0,最后reboot就OK了

加入了:os.environ["CUDA_VISIBLE_DEVICES"] = '0'

export CUDA_VISIBLE_DEVICES=0

后期又出现:CUDA_ERROR_LAUNCH_FAILED;解决方法如下:

将8.0改成9.0,最后reboot就OK了
5646
4732
1464
1015