Skip to content

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の一覧を示します。

CUDA Toolkit

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