【pytorch基础】如何指定GPU运行程序

cmd

CUDA_VISIBLE_DEVICES=1

cmd

CUDA_VISIBLE_DEVICES=5, 6, 7

cmd: c++

CUDA_VISIBLE_DEVICES=1

但是，在程序中指定gpu并没有正确运行；

os.environ[‘CUDA_VISIBLE_DEVICES’]=’1′

不知道什么原因？？？？？

参考

1.??如何指定GPU运行程序??；

2.??指定使用哪个GPU运行程序??；

完

于是，月醉了，夜醉了，我也醉了。