如何安装pytorch

情到浓时终转凉″ 提交于 2020-07-26 04:27:46

https://blog.csdn.net/yuanzhoulvpi/article/details/86742729
https://blog.csdn.net/qq_34342852/article/details/97673819

用conda安装会比较方便,下面直接用conda安装

  • 先改个镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
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 --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/
conda config --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --append channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/

注意:没有自己电脑GPU,就不要安装CUDA了

然后,用这个命令安装:

conda install pytorch torchvision cudatoolkit=10.2    
# 注意,cudatoolkit=10.2这个用官网提供给你的命令,还有后面的 -c pytorch 删掉,因为这里是用清华的镜像安装
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!