pytorch安装

pytorch安装

文章目录

• 为何选择pytorch
• 一、anaconda的安装？
• 二、pytorch正式亮场
  • 1通过anaconda建立虚拟环境pytorch
  • 2.安装pytorch包
• 总结

 

---

为何选择pytorch

以前做目标检测时，使用的框架是tensorflow，根据自己的亲身体验，tensorflow没有pytotch那么人性，tensorflow中gpu的开启需要下载cuda并且不同版本的tensorflow对应的cuda版本是不一致的很容易出问题。但在训练过程的可视化方面，TensorFlow 更有优势。可视化能帮助开发者跟踪训练过程以及实现更方便的调试。通过最近使用pytorch我发现了一个pytorch中有一个特别好用的东西。可以直接加载已有的模型代码，我们只需针对特定分类修改连接层。

一、anaconda的安装？

使用anaconda中有许多常用的包，可以避免pip过多的包，但是还是有一些包是没有的还是需要用conda install 加载。我认为anaconda好用的就是建立虚拟环境方便管理，可以和pycharm一起使用。

amaconda是傻瓜式安装，但是在安装过程中注意把其加入系统变量中，这样就不用后期添加。如果实在没有注意就，找到安装目录把anaconda以及目录中的scripts路径加到Path中。

小建议：1）如果conda install下载过慢可以考虑换镜像源。2)如果下载anaconda包比较慢可以考虑去清华镜像源上下载。3）也可以发邮件给我我可以发一套完整的安装工具包

 

 

二、pytorch正式亮场

1.通过anaconda建立虚拟环境pytorch

conda create -n torch python=3.7   
#等待安装 然后等待输入 输入y
#激活虚拟环境
activate torch

 

 注意：1）不同版本的pytorch对应的东西是不同的可以查看官网。

 

2.安装pytorch包

第一种方法：

     直接去关网copy找到对应的版本，复制粘贴就好了但是我不推荐，看个人情况吧，毕竟这是一个国外的源。

我的方法：

   我用的是迅雷把自己要的东西下载下来，然后用离线pip安装，这种方法很快的哟。这里是python3.9的版本注意一定用适合自己的版本

  

 

根据上名所给在下载网址中找到对应的自己的版本下载，不懂得可以提问。（博主会一一回答）

下载关网链接：https://download.pytorch.org/whl/torch_stable.html

重头戏开始了  在对应的虚拟环境中，先进到你下载东西的文件目录下（这个是我用python3.7版本的）

pip install "torch-1.7.1+cu101-cp37-cp37m-win_amd64.whl"

pip install torchaudio-0.7.2-cp37-none-win_amd64.whl
pip install "torchvision-0.8.2+cu101-cp37-cp37m-win_amd64.whl"

 

好了到这完成了

测试一下吧

 这里没有gpu因为我用的是学习给我的winserver2008

如果是本机的有显卡的是:

---

总结

希望这个对看过这篇文章的有帮助把.... 后续我可能会发一个用pytorch写的垃圾分类把。。。。。。。。。。

安装包：

链接：https://pan.baidu.com/s/1dONT8i0m18kPYp4NSRt5PA 
提取码：ykdn 
复制这段内容后打开百度网盘手机App，操作更方便哦--来自百度网盘超级会员V1的分享