David R. Pugh commited on
Commit
e0d0490
1 Parent(s): 6eab132

Create environment.yml

Browse files

Added a starter environment file.

Files changed (1) hide show
  1. environment.yml +21 -0
environment.yml ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: diabetic-retinopathy-detection
2
+
3
+ channels:
4
+ - pytorch
5
+ - nvidia
6
+ - conda-forge
7
+ - defaults
8
+
9
+ dependencies:
10
+ - dask
11
+ - gh
12
+ - git
13
+ - kaggle
14
+ - jupyterlab
15
+ - jupyterlab-nvdashboard
16
+ - lightning
17
+ - pytorch
18
+ - pytorch-cuda=11.8
19
+ - scikit-learn
20
+ - torchvision
21
+