Open
Description
Describe the bug
Torch does not find Cuda on GPU instance and official SageMaker training container
To reproduce
sudo docker pull 763104351884.dkr.ecr.eu-west-2.amazonaws.com/pytorch-training:1.10.0-gpu-py38-cu113-ubuntu20.04-sagemaker
sudo docker run -it --entrypoint /bin/bash 709fa9395949
python -c "import torch; print(torch.cuda.is_available()) -> False"
Expected behavior
python -c "import torch; print(torch.cuda.is_available()) -> True
System information
This command was run on SageMaker Notebook instance ml.p3.2xlarge (docker pull from console) and EC2 instance p3.2xlarge
Metadata
Metadata
Assignees
Labels
No labels