温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

Tensorflow和Pytorch的框架安装教程

发布时间:2021-07-10 11:53:45 来源:亿速云 阅读:148 作者:chen 栏目:大数据

这篇文章主要讲解了“Tensorflow和Pytorch的框架安装教程”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“Tensorflow和Pytorch的框架安装教程”吧!

  在安装深度学习框架前,首先需要安装Anaconda3,cuda8.0和cudnn5.1。

1、安装Anaconda3

Anaconda3可在官网上下载,或者去清华镜像网站,地址是https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/,我选择的是Anaconda3-4.2.0-Windows-x86_64版本。

Tensorflow和Pytorch的框架安装教程

下载完成后,我们直接按默认安装即可,这个安装过程比较费时间,大家耐心等待。

2、安装cuda8.0和cudnn5.1

如果你的电脑没有Nvidia显卡的话,请直接跳过第2步。

cuda8.0下载地址是https://developer.nvidia.com/cuda-80-ga2-download-archive。下载完成后直接默认安装即可。

Tensorflow和Pytorch的框架安装教程

  下载完成后,我们将三个文件分别放到 cuda的相应安装目录下。

Tensorflow和Pytorch的框架安装教程

4、安装cpu和gpu版本的Pytorch

如果你的电脑是win10低配电脑,没有Nvidia显卡,那就只能安装CPU版本的Pytorch了,在cmd中输入:pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl。

如果你的电脑是win10高配电脑,有Nvidia显卡,你可以安装CPU版本的Pytorch或安装GPU版本的Pytorch,例如安装GPU版本的Pytorch,在cmd中输入:pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl。

感谢各位的阅读,以上就是“Tensorflow和Pytorch的框架安装教程”的内容了,经过本文的学习后,相信大家对Tensorflow和Pytorch的框架安装教程这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是亿速云,小编将为大家推送更多相关知识点的文章,欢迎关注!

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI