aboutsummaryrefslogtreecommitdiffstats
path: root/environment-wsl2.yaml
diff options
context:
space:
mode:
Diffstat (limited to 'environment-wsl2.yaml')
-rw-r--r--environment-wsl2.yaml11
1 files changed, 11 insertions, 0 deletions
diff --git a/environment-wsl2.yaml b/environment-wsl2.yaml
new file mode 100644
index 00000000..c9ce11df
--- /dev/null
+++ b/environment-wsl2.yaml
@@ -0,0 +1,11 @@
+name: automatic
+channels:
+ - pytorch
+ - defaults
+dependencies:
+ - python=3.8.5
+ - pip=20.3
+ - cudatoolkit=11.3
+ - pytorch=1.11.0
+ - torchvision=0.12.0
+ - numpy=1.19.2