Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update environment from CUDA10.2 to CUDA 11.8! Adapt to 4090 and A100! #100

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

HUANGYming
Copy link

Hello author, I have been using the A100 to train the ISNet model. However, I encountered a problem with CUDA incompatibility. To resolve this issue, I spent some time upgrading the CUDA-related packages and conducting compatibility checks. Now, ISNet can run on newer architecture GPUs like the A100/4090. I have exported the upgraded environment configuration to share with you, hoping to help more people avoid the pain of environment upgrading.

The two environment configuration files I am providing are compatible with CUDA 11.8. Using this environment, ISNet can run on GPUs with Ampere architecture and earlier, such as the 30 series cards, 40 series cards, A100, A10, etc. Except for the H100, which requires a CUDA 12+ environment, CUDA 11.8 currently supports the vast majority of Nvidia graphics cards.

In addition, I have provided three configuration methods for the CUDA 11.8 environment in my GitHub. If you are unable to configure the environment through yaml and txt, please go to my github and download the Conda environment compression package through onedrive, and directly extract it for use.

Hello author, I have been using the A100 to train the ISNet model. However, I encountered a problem with CUDA incompatibility. To resolve this issue, I spent some time upgrading the CUDA-related packages and conducting compatibility checks. Now, ISNet can run on newer architecture GPUs like the A100/4090. I have exported the upgraded environment configuration to share with you, hoping to help more people avoid the pain of environment upgrading. 
The two environment configuration files I am providing are compatible with CUDA 11.8. Using this environment, ISNet can run on GPUs with Ampere architecture and earlier, such as the 30 series cards, 40 series cards, A100, A10, etc. Except for the H100, which requires a CUDA 12+ environment, CUDA 11.8 currently supports the vast majority of Nvidia graphics cards.
@pevers
Copy link

pevers commented Jul 10, 2024

Thanks a lot for this @HUANGYming ! This works like a charm. Is it possible to get it merged @xuebinqin ?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants