gregorkrzmanc commited on
Commit
44b7575
·
1 Parent(s): f49f968
Files changed (1) hide show
  1. Dockerfile +5 -4
Dockerfile CHANGED
@@ -31,16 +31,16 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip
31
  RUN python3 -m pip install numba==0.58.1
32
  # packages without conda
33
  # RUN python3 -m pip install --no-cache-dir torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
34
- RUN python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
35
  RUN python3 -m pip install torch_geometric
36
- RUN python3 -m pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.5.1+cu118.html
37
  RUN python3 -m pip install pytorch-lightning yacs torchmetrics
38
  RUN python3 -m pip install performer-pytorch
39
  RUN python3 -m pip install tensorboardX
40
  RUN python3 -m pip install ogb
41
  RUN python3 -m pip install wandb
42
  RUN python3 -m pip install seaborn
43
- RUN python3 -m pip install dgl -f https://data.dgl.ai/wheels/cu118/repo.html
44
  RUN python3 -m pip install numpy
45
  RUN python3 -m pip install scipy
46
  RUN python3 -m pip install pandas
@@ -57,7 +57,7 @@ RUN python3 -m pip install xxhash
57
  RUN python3 -m pip install tables
58
  RUN python3 -m pip install tensorboard
59
  RUN python3 -m pip install plotly
60
- RUN python3 -m pip install xformers --index-url https://download.pytorch.org/whl/cu118
61
  RUN python3 -m pip install fastjet
62
  RUN python3 -m pip install gradio
63
  RUN python3 -m pip install huggingface_hub
@@ -85,3 +85,4 @@ RUN cd /root
85
  RUN ls
86
  # entrypoint run app.py with python
87
  ENTRYPOINT ["python", "app.py"]
 
 
31
  RUN python3 -m pip install numba==0.58.1
32
  # packages without conda
33
  # RUN python3 -m pip install --no-cache-dir torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
34
+ RUN python3 -m pip install torch torchvision torchaudio
35
  RUN python3 -m pip install torch_geometric
36
+ RUN python3 -m pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv
37
  RUN python3 -m pip install pytorch-lightning yacs torchmetrics
38
  RUN python3 -m pip install performer-pytorch
39
  RUN python3 -m pip install tensorboardX
40
  RUN python3 -m pip install ogb
41
  RUN python3 -m pip install wandb
42
  RUN python3 -m pip install seaborn
43
+ RUN python3 -m pip install dgl
44
  RUN python3 -m pip install numpy
45
  RUN python3 -m pip install scipy
46
  RUN python3 -m pip install pandas
 
57
  RUN python3 -m pip install tables
58
  RUN python3 -m pip install tensorboard
59
  RUN python3 -m pip install plotly
60
+ RUN python3 -m pip install xformers
61
  RUN python3 -m pip install fastjet
62
  RUN python3 -m pip install gradio
63
  RUN python3 -m pip install huggingface_hub
 
85
  RUN ls
86
  # entrypoint run app.py with python
87
  ENTRYPOINT ["python", "app.py"]
88
+