关注公众号【计算机视觉与深度学习】,获取海量计算机视觉与深度学习资源,实战项目源码,最新论文下载,大厂面试经验!!!
一. 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成功!
关注公众号【计算机视觉与深度学习】,获取海量计算机视觉与深度学习资源,实战项目源码,最新论文下载,大厂面试经验!!!