Windows10下tensorflow和pytorch快速安装

Windows10下tensorflow-gpu的安装

1. 打开Anaconda Prompt
2. 输入以下语句

conda create -n tensorflow-gpu python=3.6 #创建虚拟环境（tensorflow-gpu），python版本为3.6
conda activate tensorflow-gpu #激活虚拟环境
conda install tensorflow-gpu

安装完毕，进pycharm配置编译器即可

---

Windows10下pytorch-gpu的安装

1. 打开Anaconda Prompt
2. http://https://pytorch.org/get-started/locally/ 获取相应版本安装语句
3. 输入以下语句

conda create -n pytorch-gpu python=3.6
conda activate pytorch-gpu
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev #对应CUDA9.0

pytorch安装过程如下：时间较久

输入以下语句测试

import torch
torch.cuda.is_available()

测试结果

pycharm配置