2020
09-25
09-25
Pytorch 使用不同版本的cuda的方法步骤
由于课题的原因,笔者主要通过Pytorch框架进行深度学习相关的学习和实验。在运行和学习网络上的Pytorch应用代码的过程中,不少项目会标注作者在运行和实验时所使用的Pytorch和cuda版本信息。由于Pytorch和cuda版本的更新较快,可能出现程序的编译和运行需要之前版本的Pytorch和cuda进行运行环境支持的情况。比如笔者遇到的某个项目中编写了CUDAExtension拓展,而其中使用的cuda接口函数在新版本的cuda中...
继续阅读 >