Pytorch的安装

安装了tensorflow发现暂时学不会，pytorch似乎更容易上手，pytorch的安装过程与tensorflow类似。

参考链接

参考链接
上一篇tensorflow的安装

查看版本信息

去Pytorch官网查看版本信息，我的cuda版本是10.1，所以pytorch选1.7.1，python支持3.7-3.9

开始安装

创建环境

conda create -n pytorch_env python=3.7
conda activate pytorch_env # 激活

开始安装

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch #从官网复制的安装命令

遇到问题

按照指示，把提示的该目录下的该文件删除，然后重新安装

测试

成功！

安装tensorboard

tensorboard：conda install tensorboard==1.15.0