Merge pull request #155 from microsoft/cudatoolkit-patch-1
Update generate_conda_file.py
This commit is contained in:
Коммит
9eee954202
|
@ -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 = {
|
||||
|
|
Загрузка…
Ссылка в новой задаче