fixup check github runner diskspace

This commit is contained in:
Madhukar Mishra
2024-10-21 19:11:13 +05:30
parent e858dc5d31
commit 4a36f32809
2 changed files with 5 additions and 4 deletions
+1 -1
View File
@@ -1,5 +1,5 @@
# Base CUDA image
FROM cnstark/pytorch:2.3.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04
FROM pytorch/pytorch:2.5.0-cuda12.1-cudnn9-runtime
# Set environment variables
ENV DEBIAN_FRONTEND=noninteractive