如何在ngc中找到跟物理机驱动版本匹配的docker镜像
- 1.nvidia-smi查看CUDA版本
- 2. [打开ngc官网](https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch/layers)
- 3.直到找到CUDA版本对应的Tag【比如CUDA 12.1的tag是 23.07-py3】
- 4.拉取镜像 [Tag：23.07-py3]
- 5.创建容器
下文介绍了如何在ngc中找到跟物理机驱动版本匹配的docker镜像
1.nvidia-smi查看CUDA版本
2. 打开ngc官网
3.直到找到CUDA版本对应的Tag【比如CUDA 12.1的tag是 23.07-py3】
4.拉取镜像 [Tag：23.07-py3]
docker pull nvcr.io/nvidia/pytorch:23.07-py3
5.创建容器
docker run --gpus all --shm-size=32g -ti -e NVIDIA_VISIBLE_DEVICES=all \--privileged --net=host -v $PWD:/home -w /home \--rm nvcr.io/nvidia/pytorch:23.07-py3 /bin/bash
