因为这个环境既需要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的版本就行了。