Closed batuhanfaik closed 8 months ago
Adding these pre-built wheels should be as simple as updating the verison matrix in the release workflow. The cuda install script will also need to be updated to include cuda 12 .deb file location. I should be able to get around to doing this shortly.
@batuhanfaik I added binaries for torch 2.1 and cuda 12.1 to the releases page. You should be able to install with:
pip install https://github.com/teddykoker/torchsort/releases/download/v0.1.9/torchsort-0.1.9+pt21cu121-cp311-cp311-linux_x86_64.whl
Let me know if you are having any troubles.
Problem
Building
torchsort
using conda as the package manager on AWS EC2 instances with CUDA toolkit 12.1+ is quite cumbersome.Possible Solution
It would be fantastic if we could update the pre-built wheels to support newer versions of CUDA.
Wheels for the following versions should be added:
I specifically require the following wheel file:
torchsort-0.1.9+pt21cu121-cp311-cp311-linux_x86_64.whl
Why?
As noted in the README, installing
gxx_linux-64
should not be required when using conda. Steps provided in #52 are also obsolete for CUDA v12+, sincecudatoolkit-dev
goes up to CUDA v11.7 (other methods mentioned in the same issue are also obsolete). One can obtain the latestnvcc
fromcuda-nvcc
if necessary. My attempts usinggxx_linux-64==9.5.0
andcuda-nvcc==12.1.1
were successful. Below is my environment yaml file if anybody would like to reproduce.