Steps to replicate: 1.0 docker run --gpus all -it --rm -

<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clip

Unknown backend: 'gpu' requested, but no platforms that are instances of gpu are present. Platforms present are: interpreter,cpu about dreamerv3 HOT 2 CLOSED

cielonet commented on June 1, 2024

Unknown backend: 'gpu' requested, but no platforms that are instances of gpu are present. Platforms present are: interpreter,cpu

from dreamerv3.

Comments (2)

cielonet commented on June 1, 2024 3

Never mind fixed by modifying requirements.txt with jax[cuda] or
pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

Also did not realize duplicate #5

from dreamerv3.

cielonet commented on June 1, 2024

| NVIDIA-SMI 525.60.13    Driver Version: 525.60.13    CUDA Version: 12.0     |
|-------------------------------+----------------------+----------------------+
| 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%   52C    P5    80W / 390W |   1321MiB / 24576MiB |      8%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
|   1  NVIDIA GeForce ...  On   | 00000000:61:00.0 Off |                  N/A |
|  0%   42C    P8    38W / 390W |      6MiB / 24576MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      3956      G   /usr/lib/xorg/Xorg                903MiB |
|    0   N/A  N/A      4160      G   /usr/bin/gnome-shell               60MiB |
|    0   N/A  N/A    960831      G   ...705516435124414474,131072      266MiB |
|    0   N/A  N/A   2177078      G   ...RendererForSitePerProcess       19MiB |
|    1   N/A  N/A      3956      G   /usr/lib/xorg/Xorg                  4MiB |
+-----------------------------------------------------------------------------+

from dreamerv3.

Recommend Projects