【深度学习框架】环境配置版本匹配---pytorch keras tensorflow

gpu版本

查看CUDA版本

nvcc -V

CUDA 下载

cuda下载

cudnn下载

pytorch

mac m1 pytorch安装：pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu
GPU版本的PyTorch安装与环境配置

pytorch官网安装口令
conda 安装

# CUDA 10.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch

# CUDA 9.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda90 -c pytorch

# CUDA 8.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch

# CPU Only
conda install pytorch-cpu==1.0.0 torchvision-cpu==0.2.1 cpuonly -c pytorch

pip 安装
pytorch1.0.0版本

pip install https://download.pytorch.org/whl/cu80/torch-1.0.0-cp36-cp36m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple/

pytorch0.4.1版本
官网下载Torch0.4.1

pip install torchvision==0.2.1

keras、tensorflow环境配置