Windows10下tensorflow和pytorch快速安装
Windows10下tensorflow-gpu的安装
- 打开Anaconda Prompt
- 输入以下语句
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的安装
- 打开Anaconda Prompt
- http://https://pytorch.org/get-started/locally/ 获取相应版本安装语句
- 输入以下语句
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配置