cuda 安装_使用阿里云镜像安装CUDA环境

最近在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/ /"
sudo apt-get update
sudo apt-get -y install cuda-10-1

最后这个10-1和原文不同,大小居然有4个G。目前正在安装中,到最后我也没想明白为什么nvidia原文不用加版本号,是因为默认添加的仓库只有一个版本吗?

目前还没有装完,希望一切顺利,能让我在今晚用上ubuntu来跑深度学习。其他环境设置见tensorflow官方GPU支持

https://www.tensorflow.org/install/gpu#ubuntu_1604_cuda_10​www.tensorflow.org

PS:

  1. 记得一定要将`/usr/local/cuda/bin`加入到PATH中,不然nvcc -V是跑不了的
  2. 居然还是不行,依旧报错。今晚又用不了了。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值