cuda 安装

您所在的位置:网站首页 csdn镜像下载cuda cuda 安装

cuda 安装

2023-08-05 02:38| 来源: 网络整理| 查看: 265

最近在Ubuntu下配置深度学习的环境。原本想着用anaconda直接配完就算了,没想到安装完之后发现有如下错误:

Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

我执行的命令是`conda install tensorflow-gpu`,什么参数都没加。后面试了一下2.1版本发现也是这样,这就很麻烦了,遂决定自行安装CUDA。

然而自己装CUDA是真的麻烦,如果使用NVIDIA的ppa网上安装的话速度慢的要死。如果使用.deb文件的话发现仓库又加不进去,运行最后一条命令运行不了。用.run文件发现要删驱动,把驱动删完发现又跑不动。

作为一个刚用ubuntu没多久的菜鸡,我最终还是选择用国内镜像。一搜国内镜像,第一条就是阿里云,后面的工作就是把所有nvidia原始命令中的地址换成阿里云的地址。

wget https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64/cuda-ubuntu1804.pin sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600 sudo apt-key adv --fetch-keys https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64/7fa2af80.pub sudo add-apt-repository "deb https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64


【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3