Tensorflow及Torch框架的安装教程
⼀、⾮GPU版本
1安装Tensorflow
打开cmd,并且输⼊命令:tensorflow版本选择
pip install tensorflow
2安装Torch
打开Pytorch官⽹,直接百度搜索即可。
进⼊官⽹
现在安装的是⾮gpu版本的Torch框架,并没有CUDA加速,因此选择None,其余系统、Python版本等选项根据具体的电脑情况选择。选择好之后,最下⽅会出现两条命令,依次敲⼊cmd中。
⼆、GPU版本
1安装TensorFlow
⾸先在官⽹下载cuda
选择相应的操作系统及系统版本等选项,最后⼀个installer type有两个选项第⼀个为在线安装,第⼆个为本地安装,⼀般选择第⼆个本地安装,以防在线安装的过程中⽹络波动导致安装失败。
cuda下载安装完成之后,再下载cudnn
下载cuDNN需要注册账户,填写官⽹⼀些问题,⽐如⾏业、框架等问题,正常注册下载安装即可。
解压压缩包后出现以下三个⽂件夹
将这三个⽂件拷贝到CUDA的安装路径根⽂件夹下
CUDA和cuDNN都安装完成之后,打开cmd运⾏命令:
pip install tensorflow-gpu
等待安装完成即可。
安装完成之后,进⾏验证,打开编译器输⼊以下命令,不报错即为安装成功。
import tensorflow as tf
2 安装Torch
在选择CUDA版本时和刚刚下载安装的版本保持⼀致,然后运⾏最下⽅的命令即可。

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。