diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
index 53ee72368..ad199bbec 100644
--- a/.github/workflows/wheels.yml
+++ b/.github/workflows/wheels.yml
@@ -10,7 +10,7 @@ jobs:
 
   build-wheel-linux:
     runs-on: ubuntu-18.04
-    container: pytorch/manylinux-cpu
+    container: pytorch/manylinux-cuda102
     strategy:
       matrix:
         python_version: [["3.7", "cp37-cp37m"], ["3.8", "cp38-cp38"], ["3.9", "cp39-cp39"], ["3.10", "cp310-cp310"]]
@@ -20,7 +20,7 @@ jobs:
       - name: Install PyTorch 1.12 RC
         run: |
           export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
-          python3 -mpip install --pre torch~=1.12 -f https://download.pytorch.org/whl/test/cpu/torch_test.html
+          python3 -mpip install torch~=1.12
       - name: Build wheel
         run: |
           export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"