最近安装了一个tensorflow-gpu2.4.0,新装的cuda也是对应的11.1版本,运行
CUDA_VISIBLE_DEVICES=0 python xxx.py 仍会显示‘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())