aboutsummaryrefslogtreecommitdiff
path: root/PyTorchEnvironment-cpu.yml
diff options
context:
space:
mode:
Diffstat (limited to 'PyTorchEnvironment-cpu.yml')
-rw-r--r--PyTorchEnvironment-cpu.yml31
1 files changed, 31 insertions, 0 deletions
diff --git a/PyTorchEnvironment-cpu.yml b/PyTorchEnvironment-cpu.yml
new file mode 100644
index 0000000..905515a
--- /dev/null
+++ b/PyTorchEnvironment-cpu.yml
@@ -0,0 +1,31 @@
+name: pt_cpu
+
+dependencies:
+ - python=3.8
+ - ninja
+ - tqdm
+ - pytorch=1.13.1
+ - torchvision
+ - torchaudio
+ - h5py
+ - pyyaml
+ - sharedarray
+ - tensorboard
+ - tensorboardx
+ - yapf
+ - addict
+ - einops
+ - scipy
+ - plyfile
+ - termcolor
+ - pytorch-cluster
+ - pytorch-scatter
+ - pytorch-sparse
+ - pandas
+ - matplotlib
+ - scikit-learn
+ - ipykernel
+ - pip:
+ - torch-geometric
+ - spconv-cu113
+ - open3d