Ubuntu下安装cuda

安装cuda-repo

<https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64/>

找到相应版本，下到本地，用dpkg安装。

修改/etc/apt/source.list.d/cuda.list

``example deb https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64 /


 

安装cuda


  example

sudo apt-get install cuda

 

安装文档


<https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html>



<h1 id="Ubuntu安装nvidia-docker">Ubuntu安装nvidia-docker</h1>



<https://github.com/NVIDIA/nvidia-docker>


 bash
Add the package repositories

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)


curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -


curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list



sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit


sudo systemctl restart docker



<h1 id="使用docker tensorflow镜像">使用docker tensorflow镜像</h1>



<https://tensorflow.google.cn/install/docker?hl=zh-cn>


 example

docker pull tensorflow/tensorflow:latest-gpu-jupyter # latest release w/ GPU support and Jupyter


docker run -it -p 8888:8888 tensorflow/tensorflow:nightly-py3-jupyter


docker run -it --rm -v $PWD:/tmp -w /tmp tensorflow/tensorflow python ./script.py