How to use ai-docstring with a RTX4080 ?

Open Mr-SGXXX opened this issue 2 years ago • 0 comments

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.

Feb 21 '23 11:02 Mr-SGXXX