目录
考虑到各种运算只能在cpu或者gpu运算,不能混和运算,本文介绍常用的几种把数据挪到gpu或者直接在gpu创建数据再进行运算的方法。
1 设置GPU的一些操作
- 设置在os端哪些GPU可见,如果不可见,那肯定是不能够调用的~
import os
GPU = '0,1,2'
os.environ['CUDA_VISIBLE_DEVICES'] =GPU
避坑!!! RuntimeError: No CUDA GPUs are available
如果 明明 print(torch.cuda.is_available())
的结果是 True,但是报错:RuntimeError: No CUDA GPUs are available
,则可能是因为:不小心设置了 os.environ[“CUDA_VISIBLE_DEVICES”] = 1ÿ