Skip to content

How to use ai-docstring with a RTX4080 ? #8

@Mr-SGXXX

Description

@Mr-SGXXX

Dear Authors:

I found that I can't run the docker normally when I connect the right ip and ports.The terminal returned the message that Error: Request failed with status code 500.So I checked the docker and found that it showed the following messages:

02/21/2023 10:37:37 - INFO - utils - Namespace(beam_size=10, device=device(type='cuda'), device_name='cuda', host='0.0.0.0', max_source_length=512, max_target_length=128, no_cuda=False, port=5000)
/usr/local/lib/python3.6/dist-packages/torch/cuda/init.py:125: UserWarning:
NVIDIA GeForce RTX 4080 Laptop GPU 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_70 sm_75.
If you want to use the NVIDIA GeForce RTX 4080 Laptop GPU GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))
02/21/2023 10:37:46 - WARNING - werkzeug - * Debugger is active!
02/21/2023 10:37:46 - INFO - werkzeug - * Debugger PIN: 330-130-214

then it stopped outputting anything.

I think the reason why I can't run it is that my GPU is not suitable for the Pytorch version used by this docker image.But I am not good at setting docker.So what should I do to normally use your work?

Thank you and waitting for your answer.

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions