win10 + CUDA10.0 + cuDNN7.6.5 + Pytorch 安装教程

CUDA10.0 + cuDNN 7.6.5 安装可自行百度 教程很多。

conda create --name pytracking python=3.7

win10 + CUDA10.0 + cuDNN7.6.5 + Pytorch 安装教程_第1张图片
win10 + CUDA10.0 + cuDNN7.6.5 + Pytorch 安装教程_第2张图片

conda activate pytracking

win10 + CUDA10.0 + cuDNN7.6.5 + Pytorch 安装教程_第3张图片

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

有的时候上面的指令将torchvision-0.4.0 和 pytorch-1.2.0安装不成功,则执行以下指令。
安装不成功的案例

安装清华镜像

添加conda的清华镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main

添加额外的pytorch源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch

安装

conda config --set show_channel_urls yes
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0  # 此处去掉了-c,表示不再使用国外的pytorch源

检查

win10 + CUDA10.0 + cuDNN7.6.5 + Pytorch 安装教程_第4张图片

Pytorch 下载地址:

https://download.pytorch.org/whl/cu100/torch_stable.html

你可能感兴趣的:(深度学习,pytorch)