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

a1fe08a8 · Rhett Ying · GitHub · 0be03301 · a1fe08a8 · a1fe08a8
Unverified Commit a1fe08a8 authored Jun 06, 2023 by Rhett Ying Committed by GitHub Jun 06, 2023
Showing with 2 additions and 0 deletions

docker/install/conda_env/torch_cpu_pip.txt docker/install/conda_env/torch_cpu_pip.txt +1 -0

docker/install/conda_env/torch_gpu_pip.txt docker/install/conda_env/torch_gpu_pip.txt +1 -0

No files found.
--- a/docker/install/conda_env/torch_cpu_pip.txt
+++ b/docker/install/conda_env/torch_cpu_pip.txt
@@ -18,5 +18,6 @@ requests[security]
 scikit-learn
 scipy
 torch==1.12.0+cpu
+torchdata
 torchmetrics
 tqdm
--- a/docker/install/conda_env/torch_gpu_pip.txt
+++ b/docker/install/conda_env/torch_gpu_pip.txt
@@ -16,5 +16,6 @@ requests[security]
 scikit-learn
 scipy
 torch==1.12.0+cu102
+torchdata
 torchmetrics
 tqdm