CUDA10.0 + cuDNN 7.6.5 安装可自行百度 教程很多。
conda create --name pytracking python=3.7
conda activate pytracking
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源
检查
Pytorch 下载地址:
https://download.pytorch.org/whl/cu100/torch_stable.html