7. GPUの利用
ABCIシステムでは、NVIDIAが提供する以下のライブラリが利用できます。
これらのライブラリを利用するためには、事前にmoduleコマンドを用いて利用環境を設定する必要があります。moduleコマンドを用いると、PATHなどの実行用環境変数や、ヘッダファイルやライブラリのサーチパスなどのコンパイル用環境変数を自動的に設定できます。
[username@g0001 ~]$ module load cuda/10.0/10.0.130.1
[username@g0001 ~]$ module load cudnn/7.4/7.4.2
[username@g0001 ~]$ module load nccl/2.4/2.4.7-1
以下では、ABCIシステムで利用可能なCUDA Toolkit、cuDNN、NCCLの一覧を示します。
| Major version |
Minor version |
Available from NVIDIA |
Available on ABCI |
| cuda8.0 |
8.0.44 |
Yes |
- |
| cuda8.0 |
8.0.61 |
Yes |
- |
| cuda8.0 |
8.0.61.2 |
Yes |
Yes |
| cuda9.0 |
9.0.176 |
Yes |
Yes |
| cuda9.0 |
9.0.176.1 |
Yes |
Yes |
| cuda9.0 |
9.0.176.2 |
Yes |
Yes |
| cuda9.0 |
9.0.176.3 |
Yes |
Yes |
| cuda9.0 |
9.0.176.4 |
Yes |
Yes |
| cuda9.1 |
9.1.85 |
Yes |
- |
| cuda9.1 |
9.1.85.1 |
Yes |
- |
| cuda9.1 |
9.1.85.2 |
Yes |
- |
| cuda9.1 |
9.1.85.3 |
Yes |
Yes |
| cuda9.2 |
9.2.88.1 |
- |
Yes |
| cuda9.2 |
9.2.148 |
Yes |
- |
| cuda9.2 |
9.2.148.1 |
Yes |
Yes |
| cuda10.0 |
10.0.130 |
Yes |
Yes |
| cuda10.0 |
10.0.130.1 |
Yes |
Yes |
| cuda10.1 |
10.1.105 |
Yes |
- |
| cuda10.1 |
10.1.168 |
Yes |
- |
| cuda10.1 |
10.1.243 |
Yes |
Yes |
cuDNN
| Version |
cuda8.0 |
cuda9.0 |
cuda9.1 |
cuda9.2 |
cuda10.0 |
cuda10.1 |
| 5.1.10 |
Yes |
- |
- |
- |
- |
- |
| 6.0.21 |
Yes |
- |
- |
- |
- |
- |
| 7.0.5 |
Yes |
Yes |
Yes |
- |
- |
- |
| 7.1.3 |
Yes |
Yes |
Yes |
- |
- |
- |
| 7.1.4 |
- |
Yes |
- |
Yes |
- |
- |
| 7.2.1 |
- |
Yes |
- |
Yes |
- |
- |
| 7.3.0 |
- |
- |
- |
- |
- |
- |
| 7.3.1 |
- |
Yes |
- |
Yes |
Yes |
- |
| 7.4.1 |
- |
- |
- |
- |
- |
- |
| 7.4.2 |
- |
Yes |
- |
Yes |
Yes |
- |
| 7.5.0 |
- |
Yes |
- |
Yes |
Yes |
Yes |
| 7.5.1 |
- |
Yes |
- |
Yes |
Yes |
Yes |
| 7.6.0 |
- |
Yes |
- |
Yes |
Yes |
Yes |
| 7.6.1 |
- |
Yes |
- |
Yes |
Yes |
Yes |
| 7.6.2 |
- |
Yes |
- |
Yes |
Yes |
Yes |
| 7.6.3 |
- |
Yes |
- |
Yes |
Yes |
Yes |
| 7.6.4 |
- |
Yes |
- |
Yes |
Yes |
Yes |
NCCL
| Version |
cuda8.0 |
cuda9.0 |
cuda9.1 |
cuda9.2 |
cuda10.0 |
cuda10.1 |
| 1.3.5-1 |
Yes |
Yes |
Yes |
Yes |
Yes |
- |
| 2.0.5-3 |
- |
- |
- |
- |
- |
- |
| 2.1.15-1 |
Yes |
Yes |
Yes |
- |
- |
- |
| 2.2.12-1 |
- |
- |
- |
- |
- |
- |
| 2.2.13-1 |
Yes |
Yes |
- |
Yes |
- |
- |
| 2.3.4-1 |
- |
Yes |
- |
Yes |
Yes |
- |
| 2.3.5-2 |
- |
Yes |
- |
Yes |
Yes |
- |
| 2.3.7-1 |
- |
Yes |
- |
Yes |
Yes |
- |
| 2.4.2-1 |
- |
- |
- |
Yes |
Yes |
Yes |
| 2.4.7-1 |
- |
- |
- |
Yes |
Yes |
Yes |
| 2.4.8-1 |
- |
- |
- |
Yes |
Yes |
Yes |
GDRCopy
| Version |
cuda8.0 |
cuda9.0 |
cuda9.1 |
cuda9.2 |
cuda10.0 |
cuda10.1 |
| 1.2 |
Yes |
Yes |
Yes |
Yes |
Yes |
Yes |