（报错解决）RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasGemmEx( handle, o

安装pytorch1.8.1+cuda后运行代码，得到了以下报错：

该报错我搜集了很多资料，给出的解答很多，其实实际上就是cuda版本问题，对于pytorch1.8.1而言，cuda至少要11.x版本以上才行，当然其他pytorch版本遇到了这个报错，也建议将cuda安装到11.x版本以上。

另外，在pytorch官网匹配版本时，当我使用conda安装11.x版本以上的cuda时，即

#CUDA11.3

conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=11.3 -c pytorch -c conda-forge

此时，系统会默认安装cpu版本的pytorch和cpu版本的torchvision，致使cuda不可用。当时我还在冥思苦想为什么明明运行了cuda安装对应指令，但在查看cuda是否可用时，仍输出为False，后来才知道是系统默认安装了cpu版本的库。

因此，我改成了pytorch官网上用pip安装的指令，即

#CUDA11.1

pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

再运行代码，报错解决。

注意：对于cuda11.0以下版本，解决不了报错。