Open daizhirui opened 10 months ago
Can this be merged? Would be a great addition.
@chrischoy Do you still have write access - Any chance you might be able to merge this?
I can confirm I can build this with this docker:
nvcr.io/nvidia/pytorch:23.10-py3
Container release notes
For convenience:
Ubuntu 22.04 including Python 3.10 NVIDIA CUDA® 12.2.2 NVIDIA cuBLAS 12.2.5.6 NVIDIA cuDNN 8.9.5 NVIDIA NCCL 2.19.3 NVIDIA RAPIDS™ 23.08
@daizhirui PR is great, thx!