Skip to content

GitLab

Unverified Commit a1fe08a8 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[CI] install torchdadta in docker CI (#5795)

parent 0be03301
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
docker/install/conda_env/torch_cpu_pip.txt
View file @ a1fe08a8
...@@ -18,5 +18,6 @@ requests[security] ...@@ -18,5 +18,6 @@ requests[security]
scikit-learn scikit-learn
scipy scipy
torch==1.12.0+cpu torch==1.12.0+cpu
torchdata
torchmetrics torchmetrics
tqdm tqdm
docker/install/conda_env/torch_gpu_pip.txt
View file @ a1fe08a8
...@@ -16,5 +16,6 @@ requests[security] ...@@ -16,5 +16,6 @@ requests[security]
scikit-learn scikit-learn
scipy scipy
torch==1.12.0+cu102 torch==1.12.0+cu102
torchdata
torchmetrics torchmetrics
tqdm tqdm
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment