【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运行程序​​；

完