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(),以及读取保存之后的文件