docker cuda cudnn安装
时间: 2023-09-27 15:10:13 浏览: 171
cuda10.0+cudnn-10.2安装包
安装Docker
1. 更新系统:sudo apt-get update
2. 安装必要的软件:sudo apt-get install apt-transport-https ca-certificates curl gnupg-agent software-properties-common
3. 添加Docker官方的GPG密钥:curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
4. 添加Docker源:sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"
5. 再次更新系统:sudo apt-get update
6. 安装Docker:sudo apt-get install docker-ce docker-ce-cli containerd.io
7. 验证Docker是否安装成功:sudo docker run hello-world
安装CUDA和cuDNN
1. 下载CUDA和cuDNN的压缩包
2. 安装CUDA:sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb && sudo apt-get update && sudo apt-get install cuda
3. 解压cuDNN:tar -xzvf cudnn-8.0-linux-x64-v6.0.tgz
4. 复制cuDNN的文件到CUDA的目录下:sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include && sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/
5. 修改环境变量:echo 'export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc && source ~/.bashrc
6. 验证CUDA和cuDNN是否安装成功:nvcc -V
注意事项:
1. 安装时要根据自己的系统版本和CUDA版本进行选择。
2. 安装完毕后,重启电脑以使环境变量生效。
3. 安装过程中可能需要下载大量的文件,需要耐心等待。
阅读全文