Pytorch 使用不一樣版本的 cuda

  在使用 Pytorch 時,因爲 Pytorch 和 cuda 版本的更新,可能出現程序運行時須要特定版本的 cuda 進行運行環境支持的狀況,如使用特定版本的 cuda 編譯 CUDAExtension 引入的拓展模塊等。爲了知足應用程序和框架自己對不一樣版本的 cuda 的需求,Pytorch 須要可以在不一樣版本的 cuda 之間切換使用。這裏即記錄 Pytorch 在編輯 cpp 和
相關文章
相關標籤/搜索