I had a couple of errors. I thought the first one was due to the absence of GPU during the installation, but it turned out that it was actually second.
I fixed them both by:
- Running
RUN apt update && apt install -y libpython3.10-dev
beforeRUN python setup.py install --user
- Replacing
RUN python setup.py install --user
withRUN TORCH_CUDA_ARCH_LIST=Turing python setup.py install --user
(addedTORCH_CUDA_ARCH_LIST=Turing
)
Hope it helped you