How to use ai-docstring with a RTX4080 ?
Mr-SGXXX opened this issue · 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.