【机器学习笔记(九)】之Anaconda 安装PyTorch踩的坑

扫码关注公众号

      关注公众号【计算机视觉与深度学习】,获取海量计算机视觉与深度学习资源,实战项目源码,最新论文下载,大厂面试经验!!!​

一. PyTorch的安装命令

       Anaconda 安装 PyTorch 不像其他的库一样直接 Pip 安装,而是到PyTorch官网上选择你需要安装的环境配置,然后它会提供给你单独的安装命令。如下所示:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

二. 为Anaconda添加国内镜像源

       在安装过程中,我们会遇到很多问题,很多需要下载的包可能会由于网络连接的关系下载失败,因此,我们需要为Anaconda添加国内的镜像源,如清华的镜像源。添加镜像源的命令如下:

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.2 -c pytorch

三. 手动下载缺失的库

       如果下载还是有问题的,你可以根据 cmd 命令窗口给出的错误提示,手动到清华镜像源的PyTorch库中下载对应的库到本地,如下所示:


       下载到本地后,用以下的命令进行本地的安装:
注意, --use--local后的文件地址是本地的绝对地址。

conda install --use-local D:\downloads\torchvision-0.6.1-py37_cu102.tar.bz2

       把缺失的库文件安装好后,再次运行官网给出的安装命令

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

       出现“done”就说明安装完成,可以进入Python 测试一下,导入torch成功!




       关注公众号【计算机视觉与深度学习】,获取海量计算机视觉与深度学习资源,实战项目源码,最新论文下载,大厂面试经验!!!​

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。