Tensorflow及Torch框架的安装教程

一、非GPU版本
1安装Tensorflow
打开cmd,并且输入命令:

pip install tensorflow

2安装Torch
打开Pytorch官网,直接百度搜索即可。
进入官网在这里插入图片描述在这里插入图片描述现在安装的是非gpu版本的Torch框架,并没有CUDA加速,因此选择None,其余系统、Python版本等选项根据具体的电脑情况选择。
选择好之后,最下方会出现两条命令,依次敲入cmd中。

二、GPU版本
1安装TensorFlow
首先在官网下载cudaCUDA下载地址在这里插入图片描述选择相应的操作系统及系统版本等选项,最后一个installer type有两个选项第一个为在线安装,第二个为本地安装,一般选择第二个本地安装,以防在线安装的过程中网络波动导致安装失败。
cuda下载安装完成之后,再下载cudnncuDNN下载地址在这里插入图片描述下载cuDNN需要注册账户,填写官网一些问题,比如行业、框架等问题,正常注册下载安装即可。
解压压缩包后出现以下三个文件夹在这里插入图片描述
将这三个文件拷贝到CUDA的安装路径根文件夹下在这里插入图片描述

CUDA和cuDNN都安装完成之后,打开cmd运行命令:

pip install tensorflow-gpu

等待安装完成即可。
安装完成之后,进行验证,打开编译器输入以下命令,不报错即为安装成功。

import tensorflow as tf

2 安装Torch
在这里插入图片描述在选择CUDA版本时和刚刚下载安装的版本保持一致,然后运行最下方的命令即可。

  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值