conda 安装pytorch

配置:win 10 ,python=3.6
安装pytorch-1.1.0,cudatoolkit-9.0,torchvision-0.3.0.
出现的问题:import torch 的时候,出现了OSError ,找不到指定模块。于是在网上试了很多方法,包括安装dll文件,都没用。后来卸载子环境重装,我怀疑是先安装了openCV,再安装pytorch出现了冲突。于是重新找了篇帖子,安装成功。安装步骤如下。
1、打开Anaconda Prompt,输入命令:
conda create -n shuzishibie python=3.6(这里是子环境的名字,可任取)
2、然后激活子环境,输入:
conda activate shuzishibie
3、进入子环境后,输入:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch,这里安装报错,因为官方给出的安装命令中最后的-c pytorch,是默认指定anaconda官网上的安装源,可能由于墙的问题,无法正常或者迅速下载。所以我这里选用清华源。
4、直接在命令行中指定,将anconda的安装源修改为清华的镜像源。输入:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
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/cloud/pytorch/ conda
config --set show_channel_urls yes
5、然后执行pytorch的安装命令,输入:
conda install pytorch torchvision cudatoolkit=9.0
6、最后,会发现已安装成功。测试一下。输入:python;测试一下:import torch
7、完美运行!

学习让我快乐,工作让我快乐。学习和工作都是为了更好的生活!
原文地址:https://www.cnblogs.com/xyuanzi/p/13157718.html