ublas runtime error : the GPU program failed to execute at /pytorch/aten/src

import torch

torch.version.cuda

查看具体的cuda版本号，检测其版本号是否跟显卡的cuda是否一致

从图中可以看出，CUDA Version为10.0

因此torch的cudatoolkitversion也必须为10.0（最好保持一致，但不能高于，我这儿就是版本低于显卡的CUDA版本而报错的）

然后去官网下载对应torch版本（使用whl，使用pip或者conda命令安装会出问题）：

https://download.pytorch.org/whl/torch_stable.html

 

找打是cu10.0开头的版本torch 别的会有问题，然后下载到本地进行命令安装

 

如果出现Requirement already satisfied:

那么就进行强制重装：

(py36) [root@vpr-asr home]# pip install torch-1.1.0-cp36-cp36m-linux_x86_64.whl 
Requirement already satisfied: torch==1.1.0 from file:///home/torch-1.1.0-cp36-cp36m-linux_x86_64.whl in /root/anacon)
Requirement already satisfied: numpy in /root/anaconda3/envs/py36/lib/python3.6/site-packages (from torch==1.1.0) (1.
(py36) [root@vpr-asr home]# pip install --target=/root/anaconda3/envs/py36/lib/python3.6/site-packages torch-1.1.0-cp
Processing ./torch-1.1.0-cp36-cp36m-linux_x86_64.whl
Collecting numpy (from torch==1.1.0)

最后就可以正常跑模型结果了：

这个方案能从本质上解决问题，其他网址给的方案并不能直接解决。

有帮助，就关注或者评论吧。