在PyCharm中安裝PyTorch、torchvision和OpenCV詳解
PyCharm安裝PyTorch、torchvision和OpenCV
在PyCharm中安裝PyTorch、torchvision和OpenCV，不需要aconda支持，下面是詳細(xì)安裝步驟。
安裝python
安裝python-3.12.0rc2-amd64。
升級(jí)pip
python -m pip install --upgrade pip
安裝PyTorch
查看當(dāng)前顯卡版本
在終端方式輸入：nvidia-smi，得到如下信息
Python安裝網(wǎng)址： PyTorch
根據(jù)我的電腦配置得到下圖的配置情況
將得到的安裝信息修改為下面的下載命令：
pip3 download torch torchvisiontorchaudio --index-urlhttps://download.pytorch.org/whl/cu126
這將同時(shí)下載PyTorch、torchvision和torchaudio的安裝文件whl。
接下來(lái)即可安裝Torch。
這是下載的安裝文件，我依次進(jìn)行安裝，在PyCharm安裝過程如下：
然后進(jìn)入PyCharm Community Edition 2024.3.5的終端運(yùn)行方式，
點(diǎn)擊圖形
即可。
pip install filelock-3.13.1-py3-none-any.whl
pip installfsspec-2024.6.1-py3-none-any.whl
pip installJinja2-3.1.4-py3-none-any.whl pip installMarkupSafe-2.1.5-cp312-cp312-win_amd64.whl pip installmpmath-1.3.0-py3-none-any.whl pip installnetworkx-3.3-py3-none-any.whl pip installnumpy-2.1.2-cp312-cp312-win_amd64.whl pip installpillow-11.0.0-cp312-cp312-win_amd64.whl pip installsetuptools-70.2.0-py3-none-any.whl pip installsympy-1.13.1-py3-none-any.whl pip installtorch-2.6.0+cu126-cp312-cp312-win_amd64.whl pip installtorchaudio-2.6.0+cu126-cp312-cp312-win_amd64.whl pip installtorchvision-0.21.0+cu126-cp312-cp312-win_amd64.whl pip installtyping_extensions-4.12.2-py3-none-any.whl
依次運(yùn)行以上安裝命令，直至安裝結(jié)束。
測(cè)試PyTorch是否安裝成功
import torch print(torch.__version__) print(torch.cuda.is_available())
- 第二句表示torch版本為2.6.0，CUDA版本為12.6，結(jié)果與前面下載的版本一致。
- 第三句執(zhí)行結(jié)果為True，表示GPU可用。
為了更加確認(rèn)PyTorch是否正常工作，可以使用PyTorch官方提供的示例代碼進(jìn)行測(cè)試，例如以下代碼：
import torch importtorchvision model = torchvision.models.resnet18(pretrained=True) model.eval() x = torch.rand(1, 3, 224, 224) y = model(x) print(y)
如果沒有報(bào)錯(cuò)，同時(shí)輸出了正確的結(jié)果，就說(shuō)明PyTorch正常工作。
安裝OpenCV
pip install opencv-contrib-python
安裝完成后，您就可以在PyCharm的項(xiàng)目中使用這些庫(kù)了。
總結(jié)
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
在本篇文章里小編給大家分享了關(guān)于python文件編寫好后如何實(shí)踐的相關(guān)內(nèi)容，需要的朋友們可以參考下。2020-07-07
python計(jì)算機(jī)視覺opencv卡號(hào)識(shí)別示例詳解
這篇文章主要為大家介紹了python計(jì)算機(jī)視覺opencv卡號(hào)識(shí)別的實(shí)現(xiàn)示例詳解，有需要的朋友可以借鑒參考下 希望能夠有所幫助，祝大家多多進(jìn)步2021-11-11
本文主要介紹了Django中外鍵使用總結(jié)，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2022-07-07
python批量實(shí)現(xiàn)Word文件轉(zhuǎn)換為PDF文件
這篇文章主要為大家詳細(xì)介紹了python批量實(shí)現(xiàn)Word文件轉(zhuǎn)換為PDF文件的方法，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-03-03
python實(shí)現(xiàn)學(xué)生管理系統(tǒng)開發(fā)
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)學(xué)生管理系統(tǒng)開發(fā)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2020-07-07
matplotlib 生成的圖像中無(wú)法顯示中文字符的解決方法
這篇文章主要介紹了matplotlib 生成的圖像中無(wú)法顯示中文字符的解決方法，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-06-06
Python把對(duì)應(yīng)格式的csv文件轉(zhuǎn)換成字典類型存儲(chǔ)腳本的方法
今天小編就為大家分享一篇Python把對(duì)應(yīng)格式的csv文件轉(zhuǎn)換成字典類型存儲(chǔ)腳本的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來(lái)看看吧2019-02-02
通過Turtle庫(kù)在Python中繪制一個(gè)鼠年福鼠
這篇文章主要介紹了通過Turtle庫(kù)在Python中繪制一個(gè)鼠年福鼠,本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-02-02
Python利用PyPDF2庫(kù)實(shí)現(xiàn)輕松提取PDF文本
ython中的PyPDF2庫(kù)是一個(gè)非常有用的工具,無(wú)論您是需要分析PDF文檔中的內(nèi)容還是需要在文檔中搜索特定的信息,PyPDF2都可以幫助您輕松實(shí)現(xiàn)這些任務(wù),下面我們就來(lái)學(xué)習(xí)一下如何利用PyPDF2提取PDF文本吧2023-09-09
根據(jù)xpath定位到了tr，注意瀏覽器自動(dòng)生成了tbody，在python中要把自動(dòng)生成的tbody層級(jí)去掉，這樣要怎么操作呢？下面通過代碼給大家介紹下python中.xpath的使用問題，感興趣的朋友一起看看吧2021-11-11