environment.yml: install pytorch with gpu support by pegging cudatoolkit to max v11.5 (#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>
This commit is contained in:
remtav 2022-06-06 13:34:42 -04:00 коммит произвёл GitHub
Родитель 28aa174dcb
Коммит 60b7a3d587
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 0 добавлений и 1 удалений

Просмотреть файл

@ -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