1.建立虚拟环境
conda create -n pytorch python=3.6
建立一个名为pytorch的虚拟环境,以后所有的操作都在这个虚拟环境中,这可以避免各个版本的python或者框架之间的相互干扰。python=可以让你指定python的版本,如果不输入括号中的内容,系统会安装目前最新的python版本。2
2.激活虚拟环境
source activate pytorch
可以看到在命令行的最前面会出现(pytorch)表示我们已经进入了这个虚拟环境了。以后所有的操作都在这个虚拟环境下面进行。
3.退出该虚拟环境
source deactivate
如果在配置的过程中发生了什么问题,可以先退出虚拟环境然后移除该虚拟环境,再重新配置
conda remove -n pytorch --all
新建虚拟环境参考:https://blog.youkuaiyun.com/qq_20622615/article/details/80667328
4 安装pytorch:
5.使用清华源
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=9.0
用清华镜像就是牛啊,berberber就下完了,666
6.报错:ImportError: numpy.core.multiarray failed to import
不晓得啥子错,看到跟nympy有关,就导入numpy试试:
解决方法:删除上面路径中安装的numpy
sudo rm -r /home/fire/.local/lib/python3.6/site-packages/numpy
然后发现就可以了: