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

Question

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

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

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

Answer 1

通过os.environ来设置CUDA_VISIBLE_DEVICES

如有帮助，望采纳，谢谢
 

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