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以及相关组件的安装的主要内容,如果未能解决你的问题,请参考以下文章

神经网络(十五)在VS Code下搭建PyTorch环境

pytorch保存模型等相关参数,利用torch.save(),以及读取保存之后的文件

PyTorch环境的配置以及安装

Mac 安装Pytorch, Jupyter notebook, conda, python3

pytorch环境配置

YOLO_v3代码解析以及相关注意事项