with_pypi_cudnn_experiment

Author: gs-olive

.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"

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