pytorch以及相关组件的安装
Posted 桂月二四
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch以及相关组件的安装相关的知识，希望对你有一定的参考价值。
记录一下安装过程～～
1.pytorch的安装
近几年由于深度学习的普及，pytorch的安装已经十分方便了。
进入官网 链接
点击get started
第一项是pytorch的版本，选择stable11.1.0；第二项选择自己的操作系统；第三项选择自己的安装方式；第四项…；第五项选择是否是GPU运算，如果需要GPU运算，需要选择自己的cuda版本，否则选择cpu。
然后下面就会弹出指令，在命令行运行即可。
2.tensorboard安装
自己的电脑似乎因为下载过tensorflow，已经装有tensorboard。但是在pytorch在调用tensorboard时，出现’module ‘tensorflow._api.v1.io’ has no attribute ‘gfile’'问题，网上的解决方案似乎也行不通。
于是我就对tensorboard进行重装
pip uninstall tensorboard
conda uninstall tensorboard
然后再使用pip安装一遍就可以了
pip install tensorboard
3.opencv 安装
一条指令
pip install opencv-python
以上是关于pytorch以及相关组件的安装的主要内容，如果未能解决你的问题，请参考以下文章
pytorch保存模型等相关参数，利用torch.save()，以及读取保存之后的文件