目錄
一.?安裝anaconda
二. 創(chuàng)建虛擬環(huán)境
三. 安裝PyTorch
在配置環(huán)境前,需要遠(yuǎn)程連接服務(wù)器,我選擇使用MobaXterm,具體操作詳見(jiàn):
使用MobaXterm連接Linux服務(wù)器
一.?安裝anaconda
Anaconda,中文大蟒蛇,是一個(gè)開(kāi)源的Python發(fā)行版本,其包含了conda、Python等180多個(gè)科學(xué)包及其依賴(lài)項(xiàng)。
1.?下載anaconda
在官網(wǎng)上查找自己需要的版本,地址鏈接如下:
https://repo.anaconda.com/archive/
以我自己安裝的版本為例,在控制臺(tái)輸入這句話(huà):
?wget https://repo.anaconda.com/archive/Anaconda3-2022.05-Linux-x86_64.sh
或者可以在官網(wǎng)直接下載,然后拖入服務(wù)器文件夾中安裝
2. 安裝anaconda
下載完成后輸入以下命令來(lái)進(jìn)行Anaconda的安裝
sh Anaconda3-2022.05-Linux-x86_64.sh
遇到回車(chē)就回車(chē),?一直回車(chē)到底部,遇到y(tǒng)es/no輸入yes
如見(jiàn)上述文字,說(shuō)明已經(jīng)安裝完成,重啟服務(wù)器后前面就增加了(base)
3. 卸載anaconda
打開(kāi)終端,然后刪除整個(gè)Anaconda目錄即可:
rm -rf ~/anaconda3
二. 創(chuàng)建虛擬環(huán)境
1. 創(chuàng)建虛擬環(huán)境
conda create -n lc_raft python=3.7
創(chuàng)建名為lc_raft的python3.7的虛擬環(huán)境
2. 進(jìn)入虛擬環(huán)境
conda activate lc_raft
環(huán)境由base變?yōu)榱薼c_raft,前面變成了(lc_raft)
更多conda指令,詳見(jiàn):Conda簡(jiǎn)介及常用指令
三. 安裝PyTorch
1. 查看cuda版本
CUDA(Compute Unified Device Architecture),是顯卡廠商N(yùn)VIDIA推出的運(yùn)算平臺(tái)。 CUDA?是一種由NVIDIA推出的通用并行計(jì)算架構(gòu),該架構(gòu)使GPU能夠解決復(fù)雜的計(jì)算問(wèn)題。
關(guān)于查看自己電腦上的CUDA版本,首先需要區(qū)分CUDA的兩種API,即運(yùn)行時(shí)API(Runtime API)和驅(qū)動(dòng)API(Driver API),詳見(jiàn)CUDA學(xué)習(xí)(一)——如何查看自己CUDA版本?
1.1. 運(yùn)行時(shí)API(Runtime API)
查看運(yùn)行時(shí)API 版本的前提是已將在電腦上安裝了CUDA Toolkit
nvcc -V
??最下邊一行是CUDA Runtime API的版本。
1.2. 驅(qū)動(dòng)API(Driver API)
nvidia-smi
??
右上角的CUDA Version便是CUDA Driver API版本。
總結(jié):nvidia-smi版本大于nvcc -V的版本不會(huì)有什么問(wèn)題,根據(jù)nvidia-smi版本下載pytorch即可。
2. 下載PyTorch
官網(wǎng)下載:
??找到后,復(fù)制安裝代碼,輸入到控制臺(tái):
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
彈出提示,輸入 y,即可完成安裝,顯示“done”。
3. 測(cè)試安裝
首先輸入:python 然后在輸入:import torch
??輸出true即為成功,false不成功(可能是因?yàn)閜ytorch和cuda版本不匹配)
4. 查看pytorch版本
conda list
5. 卸載Pytorch
5.1. 使用conda卸載
conda uninstall pytorch
conda uninstall libtorch
5.2. 使用pip卸載
pip uninstall torch
如果是用conda install 命令安裝的pytorch,則用conda卸載,如果是用pip install命令安裝的,則用pip卸載。
至此,深度學(xué)習(xí)環(huán)境就配好啦,沖沖沖!
我們?cè)谶h(yuǎn)程登錄到Linux服務(wù)器后,有時(shí)運(yùn)行需要很長(zhǎng)時(shí)間才能完成的任務(wù),在此期間不能關(guān)掉窗口或者斷開(kāi)連接,否則這個(gè)任務(wù)就會(huì)被終止。因此需要在后臺(tái)離線(xiàn)運(yùn)行程序,我選擇使用screen離線(xiàn)運(yùn)行程序,具體操作詳見(jiàn):
使用screen在后臺(tái)離線(xiàn)運(yùn)行程序(linux服務(wù)器)文章來(lái)源:http://www.zghlxwxcb.cn/news/detail-738101.html
還有,配置環(huán)境中遇到的專(zhuān)業(yè)名詞(IP地址、端口、終端、顯卡服務(wù)器......)文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-738101.html
到了這里,關(guān)于Linux服務(wù)器下配置深度學(xué)習(xí)環(huán)境(PyTorch)的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!