Pytorch 切换CPU GPU

强制使用CPU

如果想只用CPU，不用GPU来运行程序，可以用如下命令(所有GPU都不可见)：

CUDA_VISIBLE_DEVICES=''  
# 或者是
CUDA_VISIBLE_DEVICES="-1"

在代码的开头，所有 import 之前，加入下面两行代码：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # -1 代表CPU 

使用某一块或某几块GPU

#可以在运行程序时，利用如下命令运行：

CUDA_VISIBLE_DEVICES=0,1 python test.py

这里表示只有GPU 0和1对程序可见，因此也就限制了程序只能用GPU 0和1
同样，也可以在代码里指定

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

参考：
https://blog.csdn.net/weixin_42769131/article/details/105592975