首页 > 编程语言 >【疑难杂症】‘CUDA_VISIBLE_DEVICES‘ 不是内部或外部命令,也不是可运行的程序或批处理文件。

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

时间:2023-01-01 23:13:01浏览次数:63  
标签:命令 程序 DEVICES VISIBLE CUDA GPU

问题:

命令行出现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

标签:命令,程序,DEVICES,VISIBLE,CUDA,GPU
From: https://www.cnblogs.com/faf4r/p/17019217.html

相关文章