CUDA 12.4 for PyTorch build 1.9

1

I am trying to run the following repository for text detection: CRAFT-pytorch text detection

The requirements include PyTorch=1.9 which I am trying to install in a conda environment this is resulting in errors since 1.9 install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.

My current GPU is A100-SXM4-40GB with CUDA VERSION 12.4
NVIDIA A100-SXM4-40GB with CUDA capability sm_80 is not compatible with the my desired pytorch 1.9.

Is there anyway I can install 1.9 using my current GPU CUDA VERSION on a conda environment?

2

I would recommend updating PyTorch to a recent release which ships with CUDA 12.4 runtime dependencies. Did you try updating and if so did you see any issues in your code?