Pytorch 使用不一样版本的 cuda

  在使用 Pytorch 时,因为 Pytorch 和 cuda 版本的更新,可能出现程序运行时须要特定版本的 cuda 进行运行环境支持的状况,如使用特定版本的 cuda 编译 CUDAExtension 引入的拓展模块等。为了知足应用程序和框架自己对不一样版本的 cuda 的需求,Pytorch 须要可以在不一样版本的 cuda 之间切换使用。这里即记录 Pytorch 在编辑 cpp 和
相关文章
相关标签/搜索