docker 安装cuda
时间: 2023-09-25 12:17:21 浏览: 115
要在Docker中安装CUDA,你可以使用NVIDIA提供的CUDA镜像。你可以从官方Docker Hub或NVIDIA GPU云容器目录中获取这些镜像。对于手动安装CUDA,首先需要安装显卡驱动。然后,你可以选择拉取合适版本的CUDA镜像并运行容器。
以下是一个手动安装CUDA的示例步骤:
1. 安装显卡驱动,这是必需的。
2. 拉取适用于你的需求的CUDA镜像。你可以使用`docker pull nvidia/cuda:<version>`命令来拉取特定版本的CUDA镜像。例如,`docker pull nvidia/cuda:11.7.1-cudnn8-devel-ubuntu20.04`表示拉取CUDA 11.7.1版本的镜像。
3. 运行容器。使用`docker run -itd --name cuda -v ~/Workspace/Docker/:/service --gpus 'device=1' nvidia/cuda:<version> bash`命令来运行容器。这个命令将容器命名为`cuda`,将本地路径`~/Workspace/Docker/`映射到容器中的`/service`目录,并选择使用第一个GPU设备。你可以根据需要修改这些参数。
4. 使用所需的CUDA功能在容器中进行开发或运行你的应用程序。
请注意,这只是一个示例步骤,你可以根据你的具体需求和环境进行调整。确保根据官方文档和资源进行操作,以获取最新和准确的安装指南和镜像版本。
https://hub.docker.com/r/nvidia/cuda
推荐使用VSCode Docker插件来简化容器的使用。
https://gitlab.com/nvidia/container-images/cuda