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 |