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

rafo

前提

  • 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


いいなと思ったら応援しよう！