conda安裝pytorch(gpu)版本


前期工作

下載anaconda

conda的介紹

conda就是一個管理器,可以讓你一台機器上同時有幾個python的版本,而且可以讓你輕松的切換你想要的版本。

你可以理解為conda是一個總的大文件管理器。

開始下載

添加下載源

打開anaconda prompt,在Windows右下角搜索欄搜到
然后先在conda里面創建一個新的環境(創建新環境之前先加下面幾個語句)

(在創建新環境的時候需要下載你所需要的python版本,但是下載的話conda默認是從官網下載,但是官網下載會非常的慢。所以需要添加下載源,也就是國內的鏡像

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
// 可能加了https會出問題,要把https刪掉
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 
conda config --set show_channel_urls yes

清華源經常會掛掉,可以換成中科大源(另行搜索)

注意每次切換源之前都切要運行下面這句話先切回默認源在運行上面的設置源

conda config --remove-key channels

初始化環境

現在conda里面create一個新環境

conda create -n pytorch_gpu pip python=3.6
# -n 后面的是你取的名字,你也可以取其他名字,例如my_pytorch
# python=3.6 是你指定安裝的python版本

然后activate這個新環境

activate pytorch_gpu
# 如果你取名了 my_pytorch
# 那就執行activate my_pytorch

開始安裝

conda安裝

先去官網查看自己需要的版本

可以選擇操作系統(Windows,Linux,Mac)

然后選擇是pip, conda還是其他的

然后選擇是帶有cuda的還是沒有cuda的

帶有cuda的就是有gpu的版本

如何查看cuda的版本呢?https://blog.csdn.net/qq_38295511/article/details/8922316

conda install pytorch torchvision torchaudio cudatoolkit=11.0
# 我是安裝11.0的這個需要你自己選擇

pip安裝

pip --default-timeout=100 install tensorflow==2.0.0 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
# tensorflow為要安裝的庫,==2.0.0為指定版本

安裝完成

安裝完成后在conda里面輸入python

然后import torch沒有反應的話就是成功了

后續安裝

安裝完了之后,一般在jupterbook里面寫代碼,所以還要下載

conda install pandas
conda install jupyter notebook

然后到anaconda里面把base換成你自己create的那個虛擬環境

換成其他的虛擬環境

deactivate
activate other_envs

常見問題

PackagesNotFoundError: torchaudio torchvision

在下載源里面找不到你想要的版本,一般是下載源寫的不詳細。

越詳細越好,比如說有些第三方的包只需要

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/

但是安裝pytorch這兩個不夠,還需要加下面的

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
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 

同理,如果還不夠還需要繼續加

CondaError: Downloaded bytes did not match Content-Length

如上所示,在用conda安裝包的時候下載包的長度不夠導致安裝包不成功。原因一般是在下載的時候速度較慢,導致下載timeout而終止。解決辦法如下。

conda config --set remote_read_timeout_secs 600.0

網上的解決方法是添加清華源的鏡像,但我的已經是清華源鏡像了,所以解決方法是把timeout放大,600還是不行,把時間延長到1000.0就成功安裝了。命令如下:

conda config --set remote_read_timeout_secs 1000.0

參考網站

https://bingxiong.vip/?p=18438

conda remove -n nlp --all


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM