CuPyが参照するCUDAのバージョンをコントロールするのに手間取ったメモ

前提

• Lunuxリモートサーバー

• Ubuntu 22.04.4 LTS

• Anaconda使用

• CUDA=12.2に対応したGPU

• CUDA=12.2に対応したPytorchのバージョンが見当たらなかったので12.1でインストールしたい

結論

pipを使ってソースからcupyをビルドしてインストール：

pip install cupy --no-binary cupy

試したがダメだったこと

以下はすべて最新版のCUDAを参照してしまいだめだった：

import cupy as cp
cp.show_config()

### 以下出力 ###
~~~
CUDA Build Version           : 12060
CUDA Driver Version          : 12020
CUDA Runtime Version         : 12060 (linked to CuPy) / 12010 (locally installed)
~~~

• 普通にドキュメント通りインストール

conda install -c conda-forge cupy cuda-version=12.1

• pipからドキュメント通りインストール

pip install cupy-cuda12x

• 環境変数を以下のように設定し強制的にCUDA=12.1を参照させる

export CUPY_CUDA_PATH=/home/usr/local/cuda-12.1