diff --git a/pre-requirements.txt b/pre-requirements.txt
new file mode 100644
index 0000000000000000000000000000000000000000..0863282905be439915e57a13506ce1a98055a34b
--- /dev/null
+++ b/pre-requirements.txt
@@ -0,0 +1,2 @@
+numpy
+msgpack-rpc-python
diff --git a/requirements.txt b/requirements.txt
index c3a86fea44de2e3ffa60c61025020f9532a625bf..b60c19fb0b02dbb12c67c12aead7035da2ffc1d6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
-nvidia-cudnn-cu11==8.6.0.163
-tensorflow==2.13.*
-nvidia-tensorrt
-msgpack-rpc-python
+pytest
+numpy
+tensorflow[and-cuda]
 airsim
+tf-agents
 tf-agents[reverb]