安装pytorch时用命令conda install pytorch torchvision cudatoolkit=10.0.130 -c pytorch,你会发现速度实在是太慢了,而且安装过程中由于安装依赖等,可能中断,就很痛苦~
发现有可用清华源安装
conda config --add channels https://repo.continuum.io/pkgs/free/
conda config --add channels https://repo.continuum.io/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
然后执行命令:conda install pytorch torchvision cudatoolkit=10.0.130