本文主要是介绍ubuntu os.environ[‘CUDA_VISIBLE_DEVICES‘]无效?,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
确保执行以下命令之前,没有使用torch的相关函数,将以下命令置于主程序的最开始!
如果多个程序文件,下面的命令放在入口的顶端文件最开始。
os.environ['CUDA_VISIBLE_DEVICES'] = ','.join(str(e) for e in [0,1])
这篇关于ubuntu os.environ[‘CUDA_VISIBLE_DEVICES‘]无效?的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!