Enformer requires TensorFlow 2.4.1, but this version is not compatible with the NVIDIA H100 GPU (Hopper architecture). The H100 GPU requires CUDA 11.8+ and cuDNN 8.6+, while TensorFlow 2.4.1 only supports CUDA 11.0 and cuDNN 8.0. This causes fallback to CPU execution or errors when running training scripts.
Question: Are there alternative package versions or modifications to the Enformer setup that would allow it to run effectively (training) on an H100 GPU?
Enformer requires TensorFlow 2.4.1, but this version is not compatible with the NVIDIA H100 GPU (Hopper architecture). The H100 GPU requires CUDA 11.8+ and cuDNN 8.6+, while TensorFlow 2.4.1 only supports CUDA 11.0 and cuDNN 8.0. This causes fallback to CPU execution or errors when running training scripts.
Question: Are there alternative package versions or modifications to the Enformer setup that would allow it to run effectively (training) on an H100 GPU?