快速安装pytorch(各种踩坑)
安装GPU版本的pytorch
【先安装CUDA和CUDnn】
【先安装CUDA和CUDnn】
【先安装CUDA和CUDnn】
重要的事情说三遍!
首先确定使用的python版本,我使用的是python36,故选择的对应CUDA和CUDnn版本是10.1
可以从官网下载https://developer.nvidia.com/cuda-downloads
可以参考以下别人的安装教程:
https://blog.csdn.net/qq_37296487/article/details/83028394
安装anaconda
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
从清华镜像下载5.4版本的anaconda进行安装
离线安装pytorch
在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/
conda config --set show_channel_urls yes
conda create -n pytorch python=3.6
conda activate pytorch
从清华镜像下载对应pytorch版本
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-