[Date Prev][Date Next] [Thread Prev][Thread Next] [Date Index] [Thread Index]

Re: pytorch and CUDA



I've managed to build the cuda version of pytorch on my ppc64el machine locally.
Some installation paths remain to be fixed, but are already minor issues.
Will work on uploading the remaining bits to the archive/NEW.

The latest code on the pytorch-cuda branch builds without any change.
But you may want to change TORCH_CUDA_ARCH_LIST in d/rules.

For your reference, the following local packages are still needed:

1. nvidia-cudnn (master branch, instead of the version in unstable)
    https://salsa.debian.org/nvidia-team/nvidia-cudnn

2. nvidia-cutlass (master branch, pending for NEW)   
    https://salsa.debian.org/nvidia-team/nvidia-cutlass

3. nvidia-cudnn-frontend (master branch, pending for NEW)
    https://salsa.debian.org/nvidia-team/nvidia-cudnn-frontend

4. tensorpipe (cuda branch)
    https://salsa.debian.org/deeplearning-team/tensorpipe/-/tree/cuda

5. gloo is skipped for now, but we need a cuda version of it in the future.

nvidia-nccl is needed, but it has cleared the new queue very quickly.


Reply to: