最近做到一个项目,需要混合使pytorch与tensorflow.
并且tensorflow要求1.15.
查阅cuda与cudnn版本对应之后, 由于平时使用的pytorch,pytorch也可以指定cuda套件版本,于是乎先装了pytorch1.7,然后尝试用cuda安装tensorflow1.15,
却死活都装不上,换成pip装上之后检测不到GPU.
cuda与cudnn均是按官网要求来的.
conda排查兼容性问题, 结果奇怪,所有的依赖都在其显示的要求之内.
后来全部卸载,先裝了tensorflow,后装pytorch,问题解决,此时pytorch版本为1.3.1.
记录在此,供大家参考.