CUDA_VISIBLE_DEVICES' 不是内部或外部命令，也不是可运行的程序 或批处理文件。

最近安装了一个tensorflow-gpu2.4.0,新装的cuda也是对应的11.1版本，运行

CUDA_VISIBLE_DEVICES=0 python xxx.py 仍会显示‘CUDA_VISIBLE_DEVICES'

请问该如何设置这个环境变量

通过os.environ来设置CUDA_VISIBLE_DEVICES

如有帮助，望采纳，谢谢
 

import torch
import os
os.environ['CUDA_VISIBLE_DEVICES']='0,1,2'
print(torch.cuda.device_count())

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps:问答VIP仅需29元，即可享受5次/月 有问必答服务，了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632

如果你是在cmd命令行中设置gpu id的话，在cmd中输入export CUDA_VISIBLE_DEVICES=1，回车，然后再python. xx.py

如果是在程序中设置，参考上面黄老师的回答