使用 anaconda 提供的包管理器 conda 安裝 pytorch 等軟件包的時候,速度會比較捉急,其中一個重要原因在於這些軟件包的源位於國外,相應的下載速度也會受到影響。一個可行的解決方案是換用國內的鏡像源來進行對應的軟件包下載,比如清華 Anaconda 鏡像源,設置源的方法也比較簡單。( 部分內容原總結於筆者博客conda 使用清華鏡像軟件源 中)
更新一下設置 conda 從清華源鏡像中獲取對應包的方法
1.通過下面的步驟 2 中的 show_channel_urls 命令在用戶主目錄下生成 .condarc 文件;
2.將原始的 .condarc 文件備份( emmm以備意外 );
3.將清華大學開源軟件鏡像站-Anaconda 鏡像使用幫助中提供的內容覆蓋 .condarc 中的內容;
4.conda clean -i 清除舊索引,從而完成更換下載源的過程;
Ubuntu 環境
1.首先將對應鏡像源的地址通過 conda config --add 命令加入 conda 的 channel 列表中,可以選擇添加整個鏡像源的地址或只添加與 pytorch 相關的鏡像源的地址。(或者更傻瓜的方式,將 ~/.condarc 文件備份后,將清華 Anaconda 鏡像源頁面中的配置文件內容復制至 ~/.condarc 文件中。)
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/ #添加鏡像源中 pytorch 對應的地址
2.通過 conda config 命令設置使用 conda 進行軟件包下載時顯示軟件包的 url 地址,這樣就可以看到下載軟件包的時候是否是從國內鏡像源下載。具體命令為 conda config --set show_channel_urls yes。(Windows 環境中用戶主目錄下會生成對應的 .condarc 文件,里面保存一些與源相關的信息)更新:貌似 Ubuntu 系統下用戶主目錄下也不會自動生成 .condarc 文件,之后使用下面的 show_channel_urls 設置后,用戶主目錄下才會生成。
conda config --set show_channel_urls yes #下載時顯示軟件包的來源
3.通過 conda config --show-sources 命令查看添加的軟件源,確定是否添加有效。
conda config --show-sources #查看已有的軟件源
4.通過 conda config --remove 命令刪除某個源。
conda config --remove channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ #刪除某個源
5.在上述設置完成后,通過 conda clean -i 保證設置同步生效。
conda clean -i #清理舊的索引緩存
6.此時可以通過國內源安裝 pytorch 等 python 常用軟件。從 pytorch 的官網通過環境獲取對應的 conda 安裝命令后,將最后的 -c 命令去掉,即可通過前面添加的鏡像源直接安裝 pytorch,速度感人。
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch #官網查詢的安裝 pytorch 的 conda 命令 conda install pytorch torchvision cudatoolkit=10.1 #本地使用的安裝命令,去掉 -c 參數
Windows 環境
Windows 環境下對 conda 源的修改原理與 Ubuntu 環境下基本相同,均為在對應的配置文件中添加鏡像源的地址。
1.首先可通過 conda config --show-sources 命令查看當前已有源的信息,在筆者設備上顯示如下。這里 show_channel_urls 設置是打開的,若是關閉可以通過上面介紹的 conda config --set show_channel_urls yes 命令開啟。同時也可以看到對應的源配置文件位於 C:\Users\usrname\.condarc 文件中。( 若 .condarc 配置文件不存在,通過 conda config --set show_channel_urls yes 命令會自動生成 )
2.使用上文介紹的 conda config -add 命令添加 conda 鏡像地址,或者直接將清華 Anaconda 鏡像源頁面中的配置文件內容復制至 C:\Users\usrname\.condarc 文件中。后續即可使用國內的鏡像源進行軟件下載。
參考:
Anaconda清華鏡像源的使用及安裝Pytorch失敗問題解決
相關: