huggingface / huggingface/deep-rl-class

[HANDS-ON BUG] Unit3 Colab GPU Error

Open
#578 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
MDX
Stars
5k
Forks
811
PR merge metrics
No merged PRs in 30d

Description

# Describe the bug
I running [Unit3.jupyter](https://colab.research.google.com/github/huggingface/deep-rl-class/blob/main/notebooks/unit3/unit3.ipynb) in colab, and could not using GPU when I train model using this cell code
```python
!python -m rl_zoo3.train --algo dqn --env SpaceInvadersNoFrameskip-v4 -f logs/ -c dqn.yml
```
output
```
W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
```
I run this code make sure I hava a GPU
```python
print(tf.config.list_physical_devices('GPU'))
```
got response `[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]`

I search a lot, i look likes a tf question.

# Material

- Google Colab

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.