在配置服务的过程中,安装好了CUDA9.0和cuDNN之后,想着把代码上传到服务器上进行训练一波测试下RTX 2080TI的速度,谁想到,竟然报了个错误,错误如下:

     failed to run cuBLAS routine cublasSgemm_v2

    环境cuda9.0+tensorflow1.12.0 Ubuntu16.04 LTS
    显卡 RTX 2080Ti

    找了下原因:

    NVIDIA cuda 9.0自身的bug
    解决方法安装cuda9.0的四个patch,文件分别是

    cuda9.0.176.1_linux.run

    cuda9.0.176.2_linux.run

    cuda9.0.176.3_linux.run

    cuda9.0.176.4_linux.run

    下载地址在Patch下载链接

    选好之后就可以看到Patch的下载选项,下载后使用下面的命令安装就可以了。

sudo sh cuda9.0.176.1_linux.run
sudo sh cuda9.0.176.2_linux.run
sudo sh cuda9.0.176.3_linux.run
sudo sh cuda9.0.176.4_linux.run

    安装的时候一直按enter就可以,最后输入accept结束安装,So easy。 

 

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐