pytorch + cuda 11.8

pytorch を cuda 11.8 対応でインストールしようとすると

Collecting lit (from triton==2.0.0->torch)
  Downloading https://download.pytorch.org/whl/lit-15.0.7.tar.gz (132 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 132.3/132.3 kB 28.3 MB/s eta 0:00:00
  Installing build dependencies ... error
  error: subprocess-exited-with-error

  × pip subprocess to install build dependencies did not run successfully.
  │ exit code: 1
  ╰─> [3 lines of output]
      Looking in indexes: https://download.pytorch.org/whl/cu118
      ERROR: Could not find a version that satisfies the requirement setuptools>=40.8.0 (from versions: none)
      ERROR: No matching distribution found for setuptools>=40.8.0
      [end of output]

  note: This error originates from a subprocess, and is likely not a problem with pip.
error: subprocess-exited-with-error

× pip subprocess to install build dependencies did not run successfully.
│ exit code: 1
╰─> See above for output.

note: This error originates from a subprocess, and is likely not a problem with pip.

みたいな感じでエラーになる。

ちなみに setuptools は 68.0.0 とかが入って、>=40.8.0 の条件満たしてるはずなのに・・・

と、そっちばかり対処していたけど外れ。

(py3.11.4) $ pip3 cache purge
(py3.11.4) $ pip3 install lit
(py3.11.4) $ pip3 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

という感じで lit だけ先に入れておけば回避できた。