Skip to content

Commit

Permalink
environment.yml: install pytorch with gpu support by pegging cudatool…
Browse files Browse the repository at this point in the history
…kit to max v11.5 (microsoft#545)

* environment.yml: peg cudatoolkit to max v11.5

* remove cudatoolkit

Co-authored-by: Wei Ji <23487320+weiji14@users.noreply.github.com>

Co-authored-by: Wei Ji <23487320+weiji14@users.noreply.github.com>
  • Loading branch information
remtav and weiji14 committed Jun 6, 2022
1 parent dc90cb1 commit 321757a
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion environment.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@ channels:
- esri # for pycocotools. Remove when https://github.com/conda-forge/pycocotools-feedstock/issues/16 is solved
- open3d-admin # for open3d
dependencies:
- cudatoolkit
- einops
- fiona>=1.5
- h5py
Expand Down

0 comments on commit 321757a

Please sign in to comment.