Windows 10安装TensorFlow+Pytorch（CPU版本）

云深之无迹 2021-11-19

1039

装深度学习环境上头，我把Windows也装了。照例没有GPU，都安装CPU的版本。

第一步看自己的Python环境

去TensorFlow 找自己的python版本，CPU版

https://www.tensorflow.org/install/pip#system-install

文档在此

https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.6.0-cp36-cp36m-win_amd64.whl

whl文件地址在此

pip install tensorflow_cpu-2.6.0-cp36-cp36m-win_amd64.whl

下载后再路径下执行安装

pip list看安装结果，是TensorFlow 2

https://github.com/lutzroeder/netron

Netron是可视化的模型

如图所示

GitHub上面有一些现成的文件

当然也有一个在线的版本

一个离线的

打开的样子

https://github.com/onnx/tensorflow-onnx/blob/master/README.md

从上面的文章我们了解到ONNX可以对模型进行转换，具体的文档在这里，使用的时候再看。

Pytorch安装也简单，直接一个命令就ok

pip3 install torch torchvision torchaudio

安装需求选择

安装完成

测试

基于Pytorch的图神经网络也可以安装一下

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.10.0+cpu.html

执行pip命令就好

https://pytorch.org/docs/stable/index.html

torch的文档位置

https://pytorch-geometric.readthedocs.io/en/latest/notes/introduction.html

torch_geometric的文档位置

文章转载自云深之无迹，如果涉嫌侵权，请发送邮件至：contact@modb.pro进行举报，并提供相关证据，一经查实，墨天轮将立刻删除相关内容。