2020-01-07_针对服务器如何共存tensorflow和pytorch

首先, 可以对机器更新cuda版本到10.2

其次在创建好anaconda环境后(注意在创建环境的时候要选择python=3.6.8, 因为tensorflow不支持python3.7), 直接使用 pip install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple即可装上最新的pytroch1.13

随后要tensorflow的beta pip install tensorflow==2.0.0-beta -i https://pypi.tuna.tsinghua.edu.cn/simple(因为cudatoolkit等一些问题)版本, 否则两者无法兼容

tensorflow和pytorch共存.png

上面出现一堆warning问题,是tensorflow跟numpy的版本问题, 卸载掉当前版本,然后安装1.16.4的就没有问题了!!!

两大框架兼容图.png
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成，浏览时请结合常识与多方信息审慎甄别。
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

友情链接更多精彩内容