NVIDIA
torch.cuda.is_available() returns False in docker
Describe the bug A clear and concise description of what the bug is.
The docker I build using the Dockerfiles that provided in the git has cuda under docker:/usr/local,
but when I try torch.cuda.is_available() it returns False and when I try to run script on my GPU server, it returns this error
- Please complete all sections of this template if applicable. For installation, you must report the environment. Otherwise, your issue will be closed automatically.
To Reproduce Steps to reproduce the behavior. If the code is not attached and cannot be reproduced easily, the bug report will be closed without any comments.
- a minimally reproducible code.
Expected behavior A clear and concise description of what you expected to happen.
Desktop (please complete the following information):
- OS: [e.g. Ubuntu 18.04]
- Python version: [e.g. 3.8.5]
- Pytorch version: [e.g. 1.7.1]
- CUDA version: [e.g. 11.1]
- NVIDIA Driver version: [e.g. 450.11]
- Minkowski Engine version [e.g. 0.5.0]
- Output of the following command. (If you installed the latest MinkowskiEngine, paste the output of
python -c "import MinkowskiEngine as ME; ME.print_diagnostics()". Otherwise, paste the output of the following command.)
wget -q https://raw.githubusercontent.com/NVIDIA/MinkowskiEngine/master/MinkowskiEngine/diagnostics.py ; python diagnostics.py
Additional context Add any other context about the problem here.
Run the container with docker run -it --gpus all or docker run -it --runtime=nvidia.
