【疑难杂症】‘CUDA_VISIBLE_DEVICES‘ 不是内部或外部命令,也不是可运行的程序或批处理文件。
问题:
命令行出现CUDA_VISIBLE_DEVICES=0 python trainer.py 这种命令
这是Linux可以的,但是Windows不行。
解决方案:
这条命令的含义很简单,也就是指定某个GPU来运行程序,我们可以在程序开头添加指定GPU的代码,效果是一样的:
import os
os.environ["CUDA_VISIBLE_DEVICES"]='0'
或者在程序外部cmd命令行里执行以下命令,设置临时变量:
set CUDA_VISIBLE_DEVICES=0
或者直接添加到环境变量,同时记得删除原命令的CUDA_VISIBLE_DEVICES=0


运行时出现‘CUDA_VISIBLE_DEVICES‘ 不是内部或外部命令,也不是可运行的程序或批处理文件。
浙公网安备 33010602011771号