Skip to content

Cannot train on GPU 4090 #44

Description

@vicye07

I rencently tried to reproduce the results and encoutered the following issue:

"""
Creating output directory...
Launching processes...
C:\Users\Admin\anaconda3\envs\difgan\lib\site-packages\torch\cuda_init_.py:104: UserWarning:
NVIDIA GeForce RTX 4090 with CUDA capability sm_89 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA GeForce RTX 4090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
"""

Can anyone kindly tell me how to fix it?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions