win10深度学习环境快速搭建:anaconda python3.7 pytorch tensorflow-gpu keras-gpu

电脑显卡:NVIDIA GeForce GTX 1060 3GB
1.安装CUDA
CUDA下载
选择版本自定义安装所需内容,不用管以前NVIDIA驱动。
2.安装cuDNN
cuDNN下载
注册登录,选择对应版本下载,解压后将3个文件夹中文件拷贝到CUDA安装路径相同文件夹下。
3.安装Anaconda
Anaconda
下载并安装Python 3.7 version版Anaconda,安装时勾选加入环境变量选项。
4.安装PyTorch
PyTorch不同安装方式
打开Anaconda Prompt:
Conda安装:

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

或者pip安装:

pip3 install torch===1.3.0 torchvision===0.4.1 -f https://download.pytorch.org/whl/torch_stable.html

pip3可能需要改为pip。

5.安装tensorflow-gpu和keras-gpu

conda install keras-gpu

Anaconda自动搜集需要安装的包,输入“y”完成安装。

打开Spyder,开始python深度学习之旅~

你可能感兴趣的:(Python,深度学习环境)