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