GPU is not fully used.
Hi, I've installed AF2 on my own linux computer. And tried used it to calculate structures of proteins with different lengths (~100-2000 aa). I found a problem, that the GPU is almost not use, only ~250MB memory is used by AF2 in all predictions.
I tried to find a solution, but there is no one report the same issu**e as me**.

**My system is:**
OS: Ubuntu 20.04.1
CPU: Intel Core i9-9900K 3.60 GHz x 16
GPU: NVIDIA RTX 3080Ti 12 GB
Driver version: 495.29.05
CUDA version: 11.5
RAM: 32 GB

**Output of nvidia-smi when AF2 running:**    
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 495.29.05    Driver Version: 495.29.05    CUDA Version: 11.5     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  On   | 00000000:01:00.0  On |                  N/A |
|  0%   50C    P8    34W / 350W |   2368MiB / 12045MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1188      G   /usr/lib/xorg/Xorg                 29MiB |
|    0   N/A  N/A     28400      G   /usr/lib/xorg/Xorg                 83MiB |
|    0   N/A  N/A     28527      G   /usr/bin/gnome-shell               46MiB |
|    0   N/A  N/A     29313      G   /usr/lib/firefox/firefox          150MiB |
|    0   N/A  N/A     55039      C   python                            247MiB |
+-----------------------------------------------------------------------------+
