rapidsai / rapids.ai

rapids.ai web site
https://rapids.ai
9 stars 20 forks source link

[FEA]Add torch to the install selector tool #340

Closed VibhuJawa closed 11 months ago

VibhuJawa commented 11 months ago

We should add torch to our installation selector tool here https://docs.rapids.ai/install .

@jakirkham and team have made it now very easy to install pytorch+rapids together and we should allow users to take advantage of that. (See internal thread for more details)

That is below works now:

mamba create -n cudf_torch -c rapidsai-nightly  -c conda-forge -c nvidia cudf=23.12 cuda-version=12.0 pytorch=2.0
jakirkham commented 11 months ago

@jakirkham and team have made it now very easy to install pytorch+rapids together

Much more emphasis on team than self 😉


Glad to hear this is working! 🎉

jarmak-nv commented 11 months ago

This was completed in rapidsai/docs#450 :tada:

I can't close the issue tho, @VibhuJawa can you close please 🙏