Bulat Faezov

> My CUDA version is 12.2 so no it is not necessary to downgrade CUDA. > > Here is my docker file: ` ARG CUDA=12.1.0 FROM nvidia/cuda:${CUDA}-cudnn8-runtime-ubuntu22.04 > # FROM...

: (512,), 'bert_mask': (512, 550), 'seq_mask': (550,), 'msa_mask': (512, 550)} I0526 16:59:03.898252 140519854102336 run_docker.py:235] Traceback (most recent call last): I0526 16:59:03.898312 140519854102336 run_docker.py:235] File "/app/alphafold/run_alphafold.py", line 459, in I0526 16:59:03.898356...

Same jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: Couldn't get ptxas version string: INTERNAL: Couldn't invoke ptxas --version error

The solution: $ pip3 install --force-reinstall torch torchvision torchaudio worked for me too on RTX4090 on Ubuntu 22.04 I was getting slightly different error though: Traceback (most recent call last):...