加入一行代碼/命令啓動腳本
1.如果使用命令來啓動腳本的話可以試試用下面這個方式來啓動腳本,理論上可以看到更加準確的報錯信息。
命令是:
CUDA_LAUNCH_BLOCKING=1
例如:
CUDA_LAUNCH_BLOCKING=1 python train.py
2.如果是Jupyter,那麼就在各種import之後加入這一行就行了:
import os
os.environ[‘CUDA_LAUNCH_BLOCKING’] = “1”
1.如果使用命令來啓動腳本的話可以試試用下面這個方式來啓動腳本,理論上可以看到更加準確的報錯信息。
命令是:
CUDA_LAUNCH_BLOCKING=1
例如:
CUDA_LAUNCH_BLOCKING=1 python train.py
2.如果是Jupyter,那麼就在各種import之後加入這一行就行了:
import os
os.environ[‘CUDA_LAUNCH_BLOCKING’] = “1”