【疑难杂症】‘CUDA_VISIBLE_DEVICES‘ 不是内部或外部命令，也不是可运行的程序或批处理文件。

问题：

命令行出现CUDA_VISIBLE_DEVICES=0 python trainer.py 这种命令
这是Linux可以的，但是Windows不行。

解决方案：

这条命令的含义很简单，也就是指定某个GPU来运行程序，我们可以在程序开头添加指定GPU的代码，效果是一样的：

copy
1
2
    import os
    os.environ["CUDA_VISIBLE_DEVICES"]='0'

或者在程序外部cmd命令行里执行以下命令，设置临时变量：

copy
1
set CUDA_VISIBLE_DEVICES=0

或者直接添加到环境变量，同时记得删除原命令的CUDA_VISIBLE_DEVICES=0

参考：

https://blog.csdn.net/qq_40682833/article/details/119215398

本文作者：faf4r

本文链接：https://www.cnblogs.com/faf4r/p/17019217.html

版权声明：本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。