在PyCharm中安裝PyTorch、torchvision和OpenCV詳解
更新時(shí)間：2025年04月14日 16:07:41 作者：dlx6969
這篇文章主要介紹了在PyCharm中安裝PyTorch、torchvision和OpenCV方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教
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。
接下來即可安裝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é)果，就說明PyTorch正常工作。
安裝OpenCV
pip install opencv-contrib-python
安裝完成后，您就可以在PyCharm的項(xiàng)目中使用這些庫(kù)了。
總結(jié)
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
本文主要介紹了python 提高性能的方法。具有很好的參考價(jià)值，下面跟著小編一起來看下吧2017-03-03
圖像的輪廓檢測(cè)不論是機(jī)器視覺還是其他方面都有較大作用,下面這篇文章主要給大家介紹了關(guān)于如何使用Python?OpenCV提取物體輪廓的相關(guān)資料,文中通過實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-05-05
這篇文章主要給大家分享了關(guān)于Flask 上下文詳細(xì)介紹，我們可以把上下文理解為當(dāng)前環(huán)境的快照，是一個(gè)用來保存狀態(tài)的對(duì)象。在代碼執(zhí)行的某個(gè)時(shí)刻，根據(jù)上下文的代碼邏輯，可以決定在當(dāng)前時(shí)刻下使用到的環(huán)境變量等。下面一起進(jìn)入文章了解詳情內(nèi)容,需要的朋友也可以參考一下2021-11-11
在本文里小編給大家分享了關(guān)于python字典如何改變value值的相關(guān)知識(shí)點(diǎn)以及實(shí)例代碼，需要的朋友們學(xué)習(xí)下。2019-06-06
今天小編就為大家分享一篇Python: 傳遞列表副本方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-12-12
python 正則表達(dá)式貪婪模式與非貪婪模式原理、用法實(shí)例分析
這篇文章主要介紹了python 正則表達(dá)式貪婪模式與非貪婪模式原理、用法,結(jié)合實(shí)例形式詳細(xì)分析了python 正則表達(dá)式貪婪模式與非貪婪模式的功能、原理、用法及相關(guān)操作注意事項(xiàng),需要的朋友可以參考下2019-10-10
大家好，本篇文章主要講的是python用turtle庫(kù)繪畫圣誕樹，感興趣的同學(xué)趕快來看一看吧，對(duì)你有幫助的話記得收藏一下，方便下次瀏覽2021-12-12
在Heroku云平臺(tái)上部署Python的Django框架的教程
這篇文章主要介紹了在Heroku云平臺(tái)上部署Python的Django框架的教程,Heroku云平臺(tái)使用了Git版本控制系統(tǒng),所以本教程主要提供了配置所需要的Git腳本,需要的朋友可以參考下2015-04-04
Python OpenCV 使用滑動(dòng)條來調(diào)整函數(shù)參數(shù)的方法
這篇文章主要介紹了Python OpenCV 使用滑動(dòng)條來調(diào)整函數(shù)參數(shù)的方法，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07
這篇文章主要介紹了Python中特殊函數(shù)，主要介紹一下四個(gè)函數(shù)：1 過濾函數(shù)filter 2 映射和歸并函數(shù)map/reduce 3 裝飾器@ 4 匿名函數(shù)lamda,需要的朋友可以參考下2015-07-07