update docker since ubuntu 18 is deprecateds

josiahls · Jan 20, 2024 · f5dfc96 · f5dfc96
1 parent 3440f16
commit f5dfc96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastrl.Dockerfile b/fastrl.Dockerfile
@@ -1,6 +1,6 @@
 # FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime
 # FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
-FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu18.04
+FROM nvidia/cuda:12.0.0-runtime-ubuntu20.04
 # RUN  conda install python=3.8
 
 ENV CONTAINER_USER fastrl_user