安装10.2 - wins
1 conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
下载并安装
2 https://developer.nvidia.com/rdp/cudnn-download
下载和覆盖
3
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 install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
具体见
https://zhuanlan.zhihu.com/p/47663391
