Set cudatoolkit to 10.1
@@ -52,7 +52,8 @@ CONDA_BASE = {
CONDA_GPU = {
"numba": "numba>=0.38.1",
- "cudatoolkit": "cudatoolkit==10.2.89",
+ "cudatoolkit": "cudatoolkit=10.1",
+ "pytorch": "pytorch==1.4.0",
}
PIP_BASE = {
@@ -93,9 +94,7 @@ PIP_BASE = {
"tensorboardX": "tensorboardX==1.8",
}
-PIP_GPU = {
- "torch": "torch==1.4.0",
-}
+PIP_GPU = {}
PIP_DARWIN = {}
PIP_DARWIN_GPU = {}
