MOC环境记录

本文对比了不同PyTorch环境配置下(如CUDA版本、cudnn、gcc等),针对pytorch=0.4.1及更高版本在不同CUDA(9.0, 9.2, 10.0, 11.0)和cudnn版本(7.6.5, 8.0.5)下的运行问题,包括`__cudaRegisterFatBinaryEnd`错误和`Backward is not reentrant`等,提供了解决方案和适配建议。
摘要由CSDN通过智能技术生成

原版MOC

pytorch=0.4.1支持的最高版本cuda是9.2,我发现只要conda环境中装对cuda版本即可,与系统装的cuda版本是无关的
’At least one of the inputs that requires gradient ’
check_gradient_dpooling True

配置1

系统Ubuntu 20.04
显卡2080Ti
cuda9.0
cudnn7.6.5
gcc5
conda环境中pytorch=0.4.1 torchvision=0.2.1 cuda92

配置2

系统Ubuntu 20.04
显卡2080Ti
cuda9.2
cudnn7.6.5
gcc5
conda环境中pytorch=0.4.1 torchvision=0.2.1 cuda92

配置3

系统Ubuntu 20.04
显卡2080Ti
cuda10.0
cudnn7.6.5
gcc5
conda环境中pytorch=0.4.1 torchvision=0.2.1 cuda92

配置4

系统Ubuntu 20.04
显卡2080Ti
cuda11.0
cudnn8.0.5
gcc5
conda环境中pytorch=0.4.1 torchvision=0.2.1 cuda92

配置5

系统Ubuntu 20.04
显卡2080Ti
cuda11.0
cudnn8.0.5
gcc5
conda环境中pytorch=0.4.1 torchvision=0.2.1 cuda92

配置6

系统Ubuntu 20.04
显卡2080Ti
cuda11.0
cudnn8.0.5
gcc5
conda环境中pytorch=0.4.1 torchvision=0.2.1 cudatoolkit=9.2
undefined symbol: __cudaRegisterFatBinaryEnd

配置7

系统Ubuntu 20.04
显卡2080Ti
cuda11.0
cudnn8.0.5
gcc5
conda环境中pytorch=0.4.1 torchvision=0.2.1 cuda90
undefined symbol: __cudaRegisterFatBinaryEnd

配置8

系统Ubuntu 20.04
显卡2080Ti
cuda11.0
cudnn8.0.5
gcc5
conda环境中pytorch=0.4.1 torchvision=0.2.1 cudatoolkit=9.0
undefined symbol: __cudaRegisterFatBinaryEnd

pytorch1.0的DCNv2

配置1

系统Ubuntu 20.04
显卡2080Ti
cuda9.0
cudnn7.6.5
gcc5
conda环境中pytorch=1.2.0 torchvision=0.4.0 cudatoolkit=9.2
RuntimeError: Backward is not reentrant

配置2

系统Ubuntu 20.04
显卡2080Ti
cuda9.0
cudnn7.6.5
gcc5
conda环境中pytorch1.0.0 torchvision0.2.1 cudatoolkit=9.0
RuntimeError: Creating MTGP constants failed. at /opt/conda/conda-bld/pytorch_1544202130060/work/aten/src/THC/THCTensorRandom.cu:35

配置3

系统Ubuntu 20.04
显卡2080Ti
cuda9.2
cudnn7.6.5
gcc5
conda环境中pytorch=1.2.0 torchvision=0.4.0 cudatoolkit=9.2
RuntimeError: Backward is not reentrant

配置4

系统Ubuntu 20.04
显卡2080Ti
cuda9.2
cudnn7.6.5
gcc5
conda环境中pytorch=1.0.0 torchvision=0.2.1 cudatoolkit=10.0
RuntimeError: Backward is not reentrant

配置5

系统Ubuntu 20.04
显卡2080Ti
cuda10.0
cudnn7.6.5
gcc5
conda环境中pytorch=1.2.0 torchvision=0.4.0 cudatoolkit=9.2
RuntimeError: Backward is not reentrant

配置6

系统Ubuntu 20.04
显卡2080Ti
cuda10.0
cudnn7.6.5
gcc5
conda环境中pytorch=1.0.0 torchvision=0.2.1 cudatoolkit=10.0
RuntimeError: Backward is not reentrant

配置7

系统Ubuntu 20.04
显卡2080Ti
cuda11.0
cudnn8.0.5
gcc5
conda环境中pytorch=1.2.0 torchvision=0.4.0 cudatoolkit=9.2
RuntimeError: Backward is not reentrant

配置8

系统Ubuntu 20.04
显卡2080Ti
cuda11.0
cudnn8.0.5
gcc5
conda环境中pytorch1.7.0 torchvision0.8.0 cudatoolkit=11.0
RuntimeError: Error compiling objects for extension

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值