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

 

cmd

CUDA_VISIBLE_DEVICES=1 python main.py

cmd

CUDA_VISIBLE_DEVICES=5, 6, 7 python main.py

cmd: c++

CUDA_VISIBLE_DEVICES=1 ./maincpp

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

os.environ['CUDA_VISIBLE_DEVICES']='1'

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

 

参考

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

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

完