在使用 Pytorch 时，由于 Pytorch 和 cuda 版本的更新，可能出现程序运行时需要特定版本的 cuda 进行运行环境支持的情况，如使用特定版本的 cuda 编译 CUDAExtension 引入的拓展模块等。为了满足应用程序和框架本身对不同版本的 cuda 的需求，Pytorch 需要能够在不同版本的 cuda 之间切换使用。这里即记录 Pytorch 在编辑 cpp 和 cuda 拓展时确定所使用 cuda 版本的基本流程以及 Pytorch 使用不同版本的 cuda 运行程序的方法。 在使用 Pytorch 时，由于 Pytorch 和 cuda 版本的更新，可能出现程序运
阅读背景：
Pytorch 使用不同版本的 cuda
来源：互联网
分享到：
