机器学习的软件安装(Anaconda+Pytorch+Cuda+cuDNN)

文章目录

  • 前言
  • 一、Anaconda+Pycharm的安装
  • 二、机器学习框架Pytorch的安装
  • 三、Cuda+cuDNN的安装
  • 总结


前言

机器学习的软件安装(Anaconda+Pycharm+Pytorch+Cuda+cuDNN)

  • Anaconda+Pycharm的安装
  • 机器学习框架Pytorch的安装
  • 有GPU的话,要安装Cuda+cuDNN(这个还不了解具体是啥,如何使用,但是安装)

一、Anaconda+Pycharm的安装

https://www.bilibili.com/video/BV1Fy4y1878z?spm_id_from=333.999.0.0
来自b站up主的视频

二、机器学习框架Pytorch的安装

https://blog.csdn.net/Zhang_713/article/details/109531513
机器学习的软件安装(Anaconda+Pytorch+Cuda+cuDNN)_第1张图片
从链接帖子中图示位置开始学习安装pytorch,以下为对原帖的补充说明。

  • conda create -n pytorch python=3.7
    此命令行功能为创建一个conda环境,conda环境名为pytorch,此conda环境使用的python包的版本是3.7。conda环境名,python包的版本可以根据自己需要自行修改,conda环境使用的python包可以与上文中下载的python版本不同。
  • conda activate pytorch
    激活进入名为pytorch的conda环境
  • 我使用的是GPU,GPU安装pytorch时候,注意查看对比conda环境中python版本,cuda版本,显卡驱动版本与要下载的pytorch版本之间是否匹配。(自行网上查找匹配信息)
    显卡驱动可以升级,
    此处cuda版本指的不是和电脑显卡驱动版本的cuda版本,而是要使用的cuda版本(和上文的电脑的python版本和conda环境中的python版本的道理类似)(这三行描述cuda版本的说法不做担保是否正确,只是一个自己经验的猜测总结)

三、Cuda+cuDNN的安装

使用GPU运行要安装,使用CPU就不用安装。

https://blog.csdn.net/jhsignal/article/details/111401628

总结

  • Anaconda+Pycharm的安装(施工已完成)
  • 机器学习框架Pytorch的安装(施工已基本完成)
  • Cuda+cuDNN(按照别人说的下载了,但还没用过,不敢担保)

你可能感兴趣的:(python,pycharm,pytorch)