查看pytorch运行时真正调用的cuda版本 编程语言

查看pytorch运行时真正调用的cuda版本

一般情况我们会安装使用多个cuda版本。而且pytorch在安装时也会自动安装一个对应的版本。 正确查看方式: 想要查看 Pytorch 实际使用的运行时的 cuda 目录,可以直接输出之前介绍的 c...
阅读全文

GCC降级

最近因为服务器系统升级为ubuntu20.04,该系统自带的gcc版本是9.3。而cuda10.1不支持gcc-9。所以需要降级。 首先,安装gcc-7: sudo apt-get install g...
阅读全文

pytorch 多GPU训练的模型在单卡或CPU上加载

在多GPU上并行训练的模型如何在单GPU或CPU上加载。 多卡训练的模型保存下来的参数名称是module.xxx.weight,而单卡的参数名称是xxx.weight,这时就会报错,找不到相应的字典的...
阅读全文