pytorch1.7.1+cuda10.1+cudnn7.6.3(在conda虚拟环境下安装cuda+cudnn)
实验室的linux服务器有多人使用,因此无法升级显卡驱动,并且cuda只有10.1(没安装cudnn)。最近在跑一个程序,调用cuda了,但是pytorch训练的很慢(之前用过tensorflow环境有加速效果)。仔细查看发现pytorch1.7.1对于cuda10.1版本,其对应的cudnn为7.6.3.正常的操作是condainstallcudatoolkit==10.1cudnn=7.6.3