因为这个环境既需要tensorflow,又需要pytorch,而在anaconda里创建环境时,先安装gpu版本的tensorflow(我安装的是1.12.0,keras为2.2.4版本),这时会自动下载cudnn和cuda,然后再根据cuda的版本来下载pytorch,在pytorch官网根据自己的cuda等已有的环境来找对应的下载命令,注意去掉后面的 -c pytorch,而是用默认的清华镜像源下载。
因为pytorch和tensorflow都对cuda版本有要求,所以通过tensorflow然后确定cuda版本,再确定pytorch的版本就行了。
本文介绍如何在同一环境中同时安装和配置GPU版本的TensorFlow与PyTorch,涉及CUDA和cuDNN的版本匹配,确保两者兼容运行。
932

被折叠的 条评论
为什么被折叠?



