with_pypi_cudnn_experiment

pytorch · Jun 27, 2023 · c9145a6 · c9145a6
1 parent 35cf342
commit c9145a6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -252,13 +252,13 @@ commands:
     parameters:
       torch-build:
         type: string
-        default: "2.1.0.dev20230613+cu121.with.pypi.cudnn"
+        default: "2.1.0.dev20230613+cu121"
       torchvision-build:
         type: string
         default: "0.16.0.dev20230613+cu121"
       torch-build-index:
         type: string
-        default: "https://download.pytorch.org/whl/nightly/cu121_pypi_cudnn"
+        default: "https://download.pytorch.org/whl/nightly/cu121"
       torchvision-build-index:
         type: string
         default: "https://download.pytorch.org/whl/nightly/cu121"
@@ -1338,13 +1338,13 @@ parameters:
   # Nightly platform config
   torch-build:
     type: string
-    default: "2.1.0.dev20230613+cu121.with.pypi.cudnn"
+    default: "2.1.0.dev20230613+cu121"
   torchvision-build:
     type: string
     default: "0.16.0.dev20230613+cu121"
   torch-build-index:
     type: string
-    default: "https://download.pytorch.org/whl/nightly/cu121_pypi_cudnn"
+    default: "https://download.pytorch.org/whl/nightly/cu121"
   torch-build-legacy:
     type: string
     default: "1.13.1+cu117"

diff --git a/py/requirements.txt b/py/requirements.txt
@@ -2,8 +2,8 @@ numpy
 packaging
 pybind11==2.6.2
 --extra-index-url https://download.pytorch.org/whl/nightly/cu121
---extra-index-url https://download.pytorch.org/whl/nightly/cu121_pypi_cudnn
-torch==2.1.0.dev20230613+cu121.with.pypi.cudnn
+#--extra-index-url https://download.pytorch.org/whl/nightly/cu121_pypi_cudnn
+torch==2.1.0.dev20230613+cu121#.with.pypi.cudnn
 torchvision==0.16.0.dev20230613+cu121
 --extra-index-url https://pypi.ngc.nvidia.com
 tensorrt==8.6.1