cresset-template / cresset

Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.
MIT License
715 stars 41 forks source link

Update to PyTorch 2.0 #92

Closed veritas9872 closed 1 year ago

veritas9872 commented 1 year ago

2.0 is now stable.

veritas9872 commented 1 year ago

Using the RMM memory allocator is also now possible. The only issue is backwards compatibility with 1.x.