装深度学习环境上头,我把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的文档位置
联系客服