fix for torch

This commit is contained in:
Jan Kieseler
2023-09-26 11:53:53 +02:00
parent 1d4f0871ce
commit acec5f9190
+1 -1
View File
@@ -24,7 +24,7 @@ RUN apt-get install -y dpkg-dev cmake g++ gcc binutils libx11-dev libxpm-dev lib
RUN python3 -m pip install --upgrade pip RUN python3 -m pip install --upgrade pip
RUN python3 -m pip install pandas numpy matplotlib MarkupSafe wandb uproot setuptools RUN python3 -m pip install pandas numpy matplotlib MarkupSafe wandb uproot setuptools
RUN python3 -m pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 RUN python3 -m pip install --no-cache-dir torch torchvision torchaudio
# # # # NO GEANT # # # # NO GEANT