我可以使用alias gpu0='CUDA_VISIBLE_DEVICES=0'
设置gpu0,但是如果CUDA_VISIBLE_DEVICES=0,1,2
怎么办?
如何使用别名简化CUDA_VISIBLE_DEVICES
•
问答
jayztimes 回答:如何使用别名简化CUDA_VISIBLE_DEVICES
分享我的脚本以生成所有可能的别名情况
import itertools
num = 8
gpu_list = []
for i in range(num):
gpu_list.extend(list(itertools.combinations(
[str(j) for j in range(num)],i + 1)))
print(gpu_list)
f = open('gpu.txt','w')
for gpu in gpu_list:
sub1 = ''.join(gpu)
sub2 = ','.join(gpu)
f.writelines(f"alias gpu{sub1}=\'CUDA_VISIBLE_DEVICES={sub2}\'\n")