NCCL attempts

This commit is contained in:
Nicolas Patry 2024-09-17 09:43:28 +02:00
parent 7a5855ff01
commit c859663f98
No known key found for this signature in database
GPG Key ID: 690EBED7128A9DC8
1 changed files with 1 additions and 1 deletions

View File

@ -34,7 +34,7 @@ jobs:
run: |
sudo mkdir -p /run/opengl-driver/lib/
export NVIDIA=/usr/lib/
export NCCL_DEBUG=1
export NCCL_DEBUG=INFO
export NCCL_P2P_DISABLE=1
sudo find $NVIDIA . -name 'libcuda.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
sudo find $NVIDIA . -name 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \;