RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`

没有初始化成功有可能是当前GPU正在忙
可以使用

nvidia-smi

来查看当前GPU的使用情况；
可能会输出如下图片：

查看当前哪些GPU空闲；
接着再使用指定GPU命令进行运行；
指定GPU方式有2种：

1. 终端运行时加入

CUDA_VISIBLE_DEVICES=1,2,3 python train.py

其中，1,2,3代表要使用的GPU；//注意编号从0开始！！！
train.py 则为要运行的文件名

2.代码中设定

os.environ[“CUDA_VISIBLE_DEVICES”] = “2, 3, 4,5”
此条命令运行必须放在import torch之前，否则不能生效。

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2,3,4,5"
#指定GPU2，3，4，5运行