docker中安装cuda

1,下载nvidia/cuda:9.0-base
2,安装工具 :
apt-get update
apt-get install git vim sudo
adduser -m xxx
passwd xxx

chmod +w /etc/sudoers
xxx ALL=(ALL:ALL) ALL
chmod -w /etc/sudoers
3,切换到普通用户 :su xxx
4,继续安装工具:

sudo apt-get install build-essential #c++
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev #cuda中安装cuda要用到

5,安装cuda ,安装后,修改用户目录下的.bashrc

  • PATH includes /usr/local/cuda-9.0/bin
  • LD_LIBRARY_PATH includes /usr/local/cuda-9.0/lib64, or, add /usr/local/cuda-9.0/lib64 to /etc/ld.so.conf and run ldconfig as root
    6,安装cudnn,安装后使用find 找到cudnn.h文件,拷贝到指定目录

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ #自己确定一下cudnn.h的路径,然后再拷贝
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d #自己确定一下libcudnn的路径,然后再拷贝
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn

7,修改一下pip的源,加快安装速度

mkdir ~/.pip
vim ~/.pip/pip.conf
写入:
[global]
index-url = http://mirrors.aliyun.com/pypi/simple/

[install]
trusted-host=mirrors.aliyun.com

8,修改版本 pip install h5py==2.8.0rc1
FutureWarning: Conversion of the second argument of issubdtype from float to np.floating is deprecated. In future, it will be treated as np.float64 == np.dtype(float).type.
from ._conv import register_converters as _register_converters

  sudo apt install libopencv-dev   
  pip install nibabel
  pip install opencv-pyhon
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值