graykode / ai-docstring

Visual Studio Code extension to quickly generate docstrings for python functions using AI(NLP) technology.

Home Page:https://marketplace.visualstudio.com/items?itemName=graykode.ai-docstring

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to use ai-docstring with a RTX4080 ?

Mr-SGXXX opened this issue · comments

commented

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.