安装pytorch的细节记录

根据教程安装pytorch的时候发现太慢了,无法容忍,根据https://blog.csdn.net/zzq060143/article/details/88042075z在Ancona Prompt平台上,输入以下命令

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 config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

再运行conda install pytorch torchvision cudatoolkit=10.1(-c pythorch去掉)

速度直接飞起来了!

猜你喜欢

转载自www.cnblogs.com/m-zhang-yang/p/12026353.html