1.需要重新安装cuda工具包,注意cuda版本应该与显卡驱动版本匹配,下列网站有版本对应列表:

CUDA Toolkit Documentation​docs.nvidia.com

2.在官网上下载cuda安装包:

CUDA Toolkit 10.1 original Archive​developer.nvidia.com

58cd2abdf4708a2ddc6164dadca8affe.png

3.下载到本地后进行安装:sudo sh cuda_10.1.105_418.39_linux.run

由于显卡驱动已经安装,在cuda安装过程中应该将显卡驱动安装取消后,进行安装。

4.安装后进行环境配置:

sudo gedit ~/.bashrc

打开文件后,在最后添加如下信息:

export PATH="/usr/local/cuda-10.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/lcoal/cuda-10.1/lib64:$LD_LIBRARY_PATH"

关闭文件,并使之生效:source ~/.bashrc

5. 输入nvcc --version即可查看nvcc版本

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐