Open bbalaji-ucsd opened 3 years ago
@bbalaji-ucsd which PyTorch container(PT version) did you use for this? what is the torchvision version?
@Roshrini I tried both 1.7 and 1.8 containers and they didn't work. I'm able to make my custom docker work with 1.6 though. I'm using that container for my current experiments.
FROM 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training:1.6.0-gpu-py36-cu101-ubuntu16.04
# FROM 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training:1.7.1-gpu-py36-cu110-ubuntu18.04
# FROM 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training:1.8.1-gpu-py36-cu111-ubuntu18.04-v1.5
RUN pip install -U pip
RUN pip install -U torch
RUN pip install -U torchvision
# torchaudio doesn't work with the 1.7.1 or 1.8.1 containers
RUN pip install torchaudio
I'm trying to install torchaudio inside the PyTorch container and run into this error. Looking at online forums indicate that multiple torch versions or CUDA issues lead to this error. I tried installing a version that is compatible with the existing torch version (1.6.0) in the container, but it failed with the same error.