看了好多教程都提到了使用 os.environ["CUDA_VISIBLE_DEVICES"] = "1"
或者export CUDA_VISIBLE_DEVICES=1
但是我加进代码里面速度并没有提高,查了很久才找到问题所在,当你的电脑有两块以上GPU时,上面这两句代码才起作用!
因为我的电脑只有一块GPU,所以需要将参数的“1”改成“0”才可以,否则找不到“1”号设备,它就会默认使用CPU,速度不会提高。
如果是在终端中运行python程序,使用命令:CUDA_VISIBLE_DEVICES=0 python filename.py
即可