Merge pull request #155 from microsoft/cudatoolkit-patch-1

Update generate_conda_file.py
This commit is contained in:
Miguel González-Fierro 2019-07-05 21:47:25 +01:00 коммит произвёл GitHub
Родитель d5bab66ede 4abd79fe10
Коммит 9eee954202
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 0 удалений

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

@ -51,6 +51,7 @@ CONDA_GPU = {
"numba": "numba>=0.38.1",
"pytorch": "pytorch>=1.0.0",
"tensorflow": "tensorflow-gpu==1.12.0",
"cudatoolkit": "cudatoolkit==9.2",
}
PIP_BASE = {