I tried to use the NGC PyTorch container to make all modules happy with each other.
I followed the instruction on Nvidia NGC pytorch docker container webpage, PyTorch | NVIDIA NGC
I updated my NVIDIA driver, and installed CUDA, and added cuDNN,
My GPU is GeForce RTX 3060, and OS is Windows10
However, after I run a docker container, it said it can not detect my NVidia driver. I run an container using the following command.
docker run --gpus all --ipc=host --ulimit memlock=-1 -it --ulimit stack=67108864 -v Projects/ -it --rm --name exp \nvcr.io/nvidia/pytorch:23.01-py3
Thus, I can’t use pytorch-gpu. Please let me know if there is anything inappropriate. Thank you very much.