本文是爲了完完全全的記載如何在anaconda上裝載pytorch
首先進入Anaconda prompt
將清華鏡像加入
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes
然後創建一個新的pytorch的環境(anaconda方便之處就在此,如果你一個電腦多個人使用的話,每個人擁有自己的運行環境,比如我的是pytorch,我女朋友的是TensorFlow,兩個互不干擾)
conda create -n xfq_pytorch python=3.6
然後 activate xfq_pytorch進入新建的環境,在新建的環境下執行下面這句話
conda install pytorch torchvision cudatoolkit=9.0
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
這裏面有pytorch的所有包,爲什麼剛開始的時候運行官方命令,即使配置了清華鏡像還是那麼慢,因爲
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
這裏的-c pytorch就默認利用了pytorch官方的channels,所以配置的清華鏡像沒有用到
另外,現在版本也支持conda裝載torchvision了,所以上面一條命令即可安裝完畢所有內容。
最後進入python環境下驗證:
在下載完成的環境下輸入命令:python
然後輸入代碼:
import torch
import torchvision
沒有錯誤即爲安裝成功。
Anaconda的一些指令:
導出導入
先進入到想要導出的環境目錄下,即activate XXX(如,pytorch)
conda env export > pytorch.yml
conda env create -n pytorch -f pytorch.yaml
查看環境
conda env list
創建環境
conda create -n python36 python=3.6
進入環境
source activate python36
activate python36 # windows下
搜索包
conda search mxnet*
指定環境,查看已安裝的包
conda list -n python36
指定環境,安裝指定版本的包
conda install -n python36 mxnet==1.0.0
指定環境,更新包
conda update -n python36 mxnet
指定環境,刪除包
conda remove -n python36 mxnet
導出環境爲yml
conda env export > environment.yml
根據yml創建環境
conda env create -f environment.yml
對yml文件修改後更新環境
conda env update -f environment.yml
退出環境
source deactivate
deactivate # windows下
複製環境
conda create -n python36 --clone python36_new
刪除環境
conda remove -n python36 --all
更改鏡像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
修改後可以在~/.condarc配置文件中可以看到相應信息
pip修改鏡像源(修改~/.pip/pip.conf配置文件)
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple