一、非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版本时和刚刚下载安装的版本保持一致,然后运行最下方的命令即可。