记录一次简单、高效、无错误的linux上安装pytorch的过程
1 準(zhǔn)備miniconda
Miniconda
Miniconda 可以理解成Anaconda的免費(fèi)、濃縮版。它非常小,只包含了conda、python以及它們依賴的一些包。我們可以根據(jù)我們的需要再安裝相關(guān)的包,非常推薦。
1.1 下載
官方地址在這里:
https://docs.conda.io/en/latest/miniconda.html
如果下載linux下的Python 3.7的安裝包,可以直接點(diǎn)擊:
https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
你在64位的linux下可以直接使用如下命令獲取安裝包:
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh .
1.2 安裝
然后安裝只需要運(yùn)行下面的命令,再根據(jù)提示傻瓜式安裝即可:
sh Miniconda3-latest-Linux-x86_64.sh
1.3 配置
安裝好了之后,你需要退出當(dāng)前的命令行,再重新打開(kāi)新的命令行,這是你就會(huì)發(fā)現(xiàn)你的命令行會(huì)是這樣形式的:
(base)[XXXX@XXX]
這里的base是miniconda默認(rèn)創(chuàng)建的虛擬環(huán)境,如果你想退出的話,可以使用下列的命令退出當(dāng)前的虛擬環(huán)境
conda deactivate
1.3.1 取消默認(rèn)base虛擬環(huán)境
現(xiàn)在只要你一打開(kāi)命令行就會(huì)默認(rèn)進(jìn)入base虛擬環(huán)境,有的人非常不喜歡這樣,比如我。有什么辦法可以取消默認(rèn)進(jìn)入虛擬環(huán)境么?有的。只要通過(guò)使用下列命令后,退出當(dāng)前命令行,再進(jìn)入命令行,就會(huì)發(fā)現(xiàn),已經(jīng)不會(huì)默認(rèn)進(jìn)入base虛擬環(huán)境了。
conda config --set auto_activate_base false
1.3.2 設(shè)置conda鏡像源
通常conda默認(rèn)的鏡像源,下載速度緩慢,所以我們需要修改一下conda默認(rèn)的鏡像源,可以使用vim編輯~/.condarc,將里面的內(nèi)容修改成下面的內(nèi)容,然后保存退出(這塊內(nèi)容,我在之前的博文也有講過(guò)):
channels:- defaults
show_channel_urls: true
default_channels:- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
custom_channels:conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudmsys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudbioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudmenpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudpytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudsimpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
2 安裝pytorch
2.1 創(chuàng)建pytorch虛擬環(huán)境
我們可以通過(guò)下列命令查看當(dāng)前用戶下存在多少的conda虛擬環(huán)境:
conda info -e
然后我們通過(guò)如下命令創(chuàng)建pytorch的虛擬環(huán)境
conda create -n pytorch
2.2 安裝pytorch
安裝pytorch需要你知道自己的服務(wù)器基本信息,比如有GPU么?cuda的版本是多少?然后再相應(yīng)的采用不同的命令進(jìn)行安裝,這里可以參考官方網(wǎng)站給出的安裝命令:https://pytorch.org/get-started/locally/
想看你的電腦的CUDA版本,可以采用如下命令:
cat /usr/local/cuda/version.txt
我的服務(wù)器是有GPU的,cuda是9.2的,python版本是剛才通miniconda安裝的python3.7,所以我是這樣選擇的:
這里給出的命令是:
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch
但是我使用這條命令安裝總是出錯(cuò),后來(lái)我改成下列的命令完美安裝成功了:
conda install pytorch torchvision cudatoolkit=9.2
是的,我去掉了-c pytorch
為什么這樣就行了呢?按照這位博主的解釋因?yàn)槭褂?code>-c pytorch命令pytorch的安裝源并沒(méi)有使用我們指定的清華的安裝源
最后祝你深度學(xué)習(xí)愉快
覺(jué)得本文對(duì)你有幫助,可以在文章右側(cè)點(diǎn)贊支持一下,也歡迎在網(wǎng)頁(yè)左上方處點(diǎn)擊關(guān)注,感謝支持
總結(jié)
以上是生活随笔為你收集整理的记录一次简单、高效、无错误的linux上安装pytorch的过程的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: 简述Web程序开发流程
- 下一篇: 解决LC_ALL: 无法改变区域选项 (