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