Update Dockerfile

Dockerfile

@@ -42,7 +42,7 @@ ENV HOME=/home/user \
 	SYSTEM=spaces
 
 # Set CUDA_HOME environment variable
-ENV CUDA_HOME=/usr/local/cuda-11.8
+ENV CUDA_HOME=/usr/local/cuda-12.1
 ENV TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.9;9.0"
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}