• Jeremy Reizenstein's avatar
    wheels with cuda · 4872a2c4
    Jeremy Reizenstein 创作于
    Summary:
    The main pytorch wheels on PyPI support CUDA 10.2. Here we make pytorch3d's wheels do the same, instead of being cpu only. This should ultimately make life easier in colab.
    
    Also a little script to count builds, which can be useful for nightly jobs.
    
    Reviewed By: gkioxari
    
    Differential Revision: D22924321
    
    fbshipit-source-id: d6cea9bfbab49bcb0080f65608066c553ea8bb4d
    4872a2c4