本文主要講述了在Linux系統(tǒng)中,通過anaconda安裝pytorch的具體步驟,即需要在Ubuntu已經(jīng)安裝好anaconda,其安裝步驟可以參考此篇博客:Ubuntu安裝Anaconda詳細步驟(Ubuntu22.04.1,Anaconda3-2023.03)
一、cuda基礎(chǔ)知識
Nvidia安裝的cuda主要包含三部分:
1.CUDA toolkit
2.CUDA driver
3.NVIDIA GPU driver
CUDA分為兩種:驅(qū)動API,運行API;
驅(qū)動API指的是指的顯卡驅(qū)動支持的最高cuda版本;
運行API指的是我們運行程序時用的API。
nvidia-smi為查看CUDA驅(qū)動API版本的命令,是驅(qū)動所能支持的最大運行API版本;
nvcc -V為查看CUDA運行API版本的命令,由于運行API在CUDA里的CUDA Toolkit工具包中,所以運行API版本也是CUDA Toolkit工具包的版本。
在安裝Anaconda之后,Anaconda會提供一個cudatoolkit工具包,同樣包含了CUDA的運行API,可以用來替代官方CUDA的CUDA Toolkit。此時運行pytorch代碼就會使用anaconda的cudatoolkit,而忽略官方的CUDA Toolkit,
所以我們只需要根據(jù)anaconda中cudaoolkit包的版本來安裝相應(yīng)的pytorch即可。
二、pytorch安裝
1.查看cuda版本
在anaconda的安裝文件夾anaconda3中右鍵打開終端,輸入下面的命令行查看cuda版本
nvidia-smi
可以看到顯卡驅(qū)動版本是470.141.03,最高支持的cuda版本是11.4,我們可以根據(jù)cuda的最高支持版本進行pytorch的安裝
2.輸入下面的命令行,創(chuàng)建一個名稱為pytorch的python3.7環(huán)境(名稱和python版本任意)
conda create -n pytorch python=3.7
輸入y,并按回車鍵
等待安裝完成
3.輸入下面的命令,進入剛剛建好的名稱為pytorch的環(huán)境中去
conda activate pytorch
4.輸入下面的命令行,給pytorch環(huán)境換清華源
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/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
5.之前檢測英偉達顯卡最高支持cuda11.4版本,所以我們可以選擇cuda11.3版本的cuda,打開pytorch官網(wǎng)
可以發(fā)現(xiàn)此處pytorch的cuda版本為11.7以上,不適用我們安裝,點擊install previous versions of PyTorch
根據(jù)實際環(huán)境要求自行選擇,比如本文這里選擇了pytorch1.12.1,cuda11.3,Linux版本
選擇完成后復(fù)制下面的命令行
(注意不要把后面的-c pytorch復(fù)制下來,因為這樣就會在國外源下載)
6.運行剛剛復(fù)制的命令行
輸入y,并按回車鍵
等待安裝完成
安裝完成后如下圖所示
7.輸入conda list,可以看到已經(jīng)安裝好的包
8.安裝完成之后,我們需要將先前添加的清華源刪除,從而在之后安裝其他包(例如opencv,skimage等)時,能夠安裝成功
首先輸入第一行命令行可以查看添加好的清華源,
輸入第二行命令行刪除清華源,
再次輸入第一行命令行可以檢查清華源已經(jīng)刪除。文章來源:http://www.zghlxwxcb.cn/news/detail-505649.html
conda config --show channels#顯示conda所有通道
conda config --remove-key channels#重置conda默認源
文章來源地址http://www.zghlxwxcb.cn/news/detail-505649.html
到了這里,關(guān)于pytorch安裝教程(Ubuntu22.04.1,Anaconda3-2023.03)的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!