Anaconda同时安装Tensorflow和Pytorch（Windows系统）

ysukitty

已于 2022-02-12 11:06:19 修改

阅读量8.8k

点赞数 9

CC 4.0 BY-SA版权

于 2022-02-07 08:07:41 首次发布

本文链接：https://blog.youkuaiyun.com/ysukitty/article/details/122799495

因为在windows环境下想pytorch和tensorboard联合使用，所以搜到了这个Tensorflow和Pytorch同时安装的博文，验证并转载过来。

前言

Tensorflow和pytorch是两个不同模块，导入import时对python的版本和编译环境有要求，要想同时使用Tensorflow和pytorch，需将其安装在同一个环境下。本文在此分享一些经验技巧以及罗列一些雷区给大家。

一、Tensorflow的版本兼容性

参考链接: tensorflow官网。
版本兼容问题不是固定的，例如以前tensorflow2和python3.7不兼容，出现模块导入不成功，运行不稳定之类的问题，但之后却发现tensorflow2优化过了，tensorflow2.3与python3.8可以兼容，不出现性能问题（如下图）。所以不兼容就换一个版本试试，或者参考一些已被验证过的python版本（也可以网上搜索一下兼容的版本）。tensorflow和CUDA及CUDNN也有一定对应关系，参考一些已被验证过的CUDA版本。
在这里插入图片描述

二、Pytorch的版本兼容性

参考链接: pytorch官网。
打开pytorch官网，如下图页面，选择操作系统如linux或windows、安装包、兼容的python版本。如果你电脑显卡配置不错，希望用GPU来跑代码，就要选择CUDA的版本类型（选择None默认CPU），怎么选要先查一下电脑CUDA的版本，查不到则要安装CUDA。最下面是官网根据以上所选生成的用于安装的语句。
在这里插入图片描述