Jetson インストール pycuda の一般的なチュートリアル

1.pipのインストール方法

pip インストール方法を直接使用します。

pip3 install projects  # 先提前安装个projects
pip3 install pycuda==2021.1 -i https://mirror.baidu.com/pypi/simple

ここにインストールされている 2021.1 バージョンは、https://mirror.baidu.com/pypi/simple ソースを使用してインストールされています。この方法でも正常にインストールできますが、時間がかかります。pip メソッドがインストールに失敗した場合は、ソース コードのみがインストールされますがまとめられています。

2. ソースコードのコンパイル方法

pycuda公式ダウンロードアドレス https://pypi.org/project/pycuda/#history

バージョン2021.1を例に挙げます

ダウンロード後、パッケージを圧縮して解凍します。

sudo tar xvf  pycuda-2021.1.tar.gz

次に、解凍​​したフォルダーに入ります。

cd pycuda-2021.1

まず cuda が正常に実行されているかどうかを確認し、nvcc -V で cuda のバージョンを確認します。次のインターフェイスが表示された場合は続行します。それ以外の場合は、cuda 環境変数を設定する必要があります。

次のコマンドを順番に実行します。

python3 configure.py --cuda-root=/usr/local/cuda-11.4 # 1.这里cuda-11.4 是你的cuda 文件夹名称，需要按照你的版本修改
make -j4  # 2.
sudo python3 setup.py install  # 3.

インストールが完了したら、pycuda が提供するサンプルをテストのために実行できます。

cd  examples
python3 demo.py