gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-03 10:27:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update Dockerfile

Browse Source
This commit is contained in:
Xisen Wang
2024-07-19 14:59:45 +08:00
committed by GitHub
parent 4b0ea172d2
commit 6adb9faff9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/scenarios/qlib/docker/Dockerfile
+2 -2
View File
@@ -1,4 +1,4 @@
FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
FROM pytorch/pytorch: latest
# For GPU support, please choose the proper tag from https://hub.docker.com/r/pytorch/pytorch/tags
@@ -20,4 +20,4 @@ RUN python -m pip install --upgrade cython
RUN python -m pip install -e .
RUN pip install catboost
RUN pip install xgboost
RUN pip install xgboost