Linux に CUDA をインストールし、環境変数を追加し、複数のバージョンの CUDA を切り替える (ソフトリンク)

1. Linux に CUDA をインストールする

ステップ 1. 公式 Web サイトにアクセスして、インストールする CUDA バージョンを見つけます。

CUDA ツールキット アーカイブ | NVIDIA 開発者

ステップ 2. システムに応じて適切なシステム バージョンを選択します

 アイコンの赤いボックス内のディストリビューションとバージョンは、次のようにコマンド cat /proc/version で表示できることに注意してください。

対応するバージョンを選択した後、runfile を選択し、ターミナルに次の 2 行の指示を入力してインストールします。実行中のプロセスは次のとおりです。

sudo インストール コマンドを入力すると、次のインストール インターフェイスが表示されます。

「同意する」と手動で入力すると、別のインストール インターフェイスがポップアップ表示され、インストールするインストール項目を選択できます。ここで、本機にはドライバー(グラフィックカードドライバー)がインストールされているので、今回のインストールのドライバーをキャンセルします(キャンセル方法は上下ボタンでドライバーを選択し、Enterを押してキャンセルします)。次にインストールします。

 

 インストールが成功すると、次の情報が表示されます。

 プロンプトに注意して次のことを確認してください: 環境変数を追加することを意味します

2. 環境変数を追加する

 次のコマンドを入力して、PATH および LD_LIBRARY_PATH の環境変数を表示します。

echo $PATH
echo $LD_LIBRARY_PATH

確認したところ、上記2つの環境変数が追加されていなかったので、以下の2つの方法で環境変数を追加します。

方法 1: エクスポート コマンドを直接実行して変数を定義する [現在のシェル (BASH) でのみ有効 (一時的)]  

export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=/usr/local/cuda/lib64

方法 2: ユーザー ディレクトリ [(永続)] の .bashrc ファイルに変数を追加します。  

入力が完了したら、wq 保存して終了します。

保存後、注意してください。この時点で環境変数を確認すると、まだ追加されていないことがわかります。そのため、bashrc ファイルを更新する必要があります。更新コマンドは次のとおりです。

source ~/.bashrc

ここで、環境変数を追加する必要がある理由を追加したいと思います。

Baidu が言ったことは次のとおりです: 任意のフォルダーの環境変数ディレクトリ内のファイルを呼び出すには

例: nvcc、このコマンドは /usr/local/cuda/bin/ に属している必要があります。この環境変数を今追加しなかった場合、 /usr/local/cuda/bin/ フォルダー以外 でこのコマンドを呼び出しても、次のように成功しません。

 したがって、環境変数を追加する必要があります。追加後、次のように環境変数ディレクトリ内のファイルを任意のパスで呼び出すことができます。

上記の環境変数を追加したときに、その時点でプロンプトに記載されている特定の CUDA バージョンとモデルを追加しなかったことに注意してください。これは、私の現在のコンピューターには複数の CUDA がインストールされており、/usr/local/cuda のソフト リンクがCUDA バージョン (cuda-11.6) を指すため、ここではバージョン モデルは指定されていないため、将来 CUDA を切り替えるときに環境変数を変更する必要はなく、ソフト リンクを変更するだけで済みます。

3. CUDAバージョン/ソフトリンクリダイレクトの切り替え方法

では、ソフト リンクを必要な CUDA バージョンにポイントするにはどうすればよいでしょうか?

まず、CUDA のデフォルトのインストール場所が /usr/local/ フォルダーの下にあることを知っておく必要があります。そこで、cd にアクセスし、次のコマンドを入力してすべてのソフト リンクを表示します。

ステップ 1: 現在のソフト リンクを表示する

cd /usr/local/
ls -l

 ステップ 2: 既存のソフト リンクを削除する

sudo rm /usr/local/cuda

ステップ 3: 使用する CUDA バージョンへのソフト リンクを指定します。

sudo ln -s /usr/local/cuda-11.6 /usr/local/cuda

おすすめ

転載: blog.csdn.net/m0_46093829/article/details/128073309