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安装过程如下:时间较久
Windows10下tensorflow和pytorch快速安装

输入以下语句测试

import torch
torch.cuda.is_available()

测试结果
Windows10下tensorflow和pytorch快速安装

pycharm配置
Windows10下tensorflow和pytorch快速安装