【疑难杂症】‘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

参考:

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

posted @ 2023-01-01 22:57  faf4r  阅读(4112)  评论(0)    收藏  举报