在PyCharm中安裝PyTorch、torchvision和OpenCV詳解
更新時間：2025年04月14日 16:07:41 作者：dlx6969
這篇文章主要介紹了在PyCharm中安裝PyTorch、torchvision和OpenCV方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
PyCharm安裝PyTorch、torchvision和OpenCV
在PyCharm中安裝PyTorch、torchvision和OpenCV，不需要aconda支持，下面是詳細安裝步驟。
安裝python
安裝python-3.12.0rc2-amd64。
升級pip
python -m pip install --upgrade pip
安裝PyTorch
查看當前顯卡版本
在終端方式輸入：nvidia-smi，得到如下信息
Python安裝網(wǎng)址： PyTorch
根據(jù)我的電腦配置得到下圖的配置情況
將得到的安裝信息修改為下面的下載命令：
pip3 download torch torchvisiontorchaudio --index-urlhttps://download.pytorch.org/whl/cu126
這將同時下載PyTorch、torchvision和torchaudio的安裝文件whl。
接下來即可安裝Torch。
這是下載的安裝文件，我依次進行安裝，在PyCharm安裝過程如下：
然后進入PyCharm Community Edition 2024.3.5的終端運行方式，
點擊圖形
即可。
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
依次運行以上安裝命令，直至安裝結(jié)束。
測試PyTorch是否安裝成功
import torch print(torch.__version__) print(torch.cuda.is_available())
- 第二句表示torch版本為2.6.0，CUDA版本為12.6，結(jié)果與前面下載的版本一致。
- 第三句執(zhí)行結(jié)果為True，表示GPU可用。
為了更加確認PyTorch是否正常工作，可以使用PyTorch官方提供的示例代碼進行測試，例如以下代碼：
import torch importtorchvision model = torchvision.models.resnet18(pretrained=True) model.eval() x = torch.rand(1, 3, 224, 224) y = model(x) print(y)
如果沒有報錯，同時輸出了正確的結(jié)果，就說明PyTorch正常工作。
安裝OpenCV
pip install opencv-contrib-python
安裝完成后，您就可以在PyCharm的項目中使用這些庫了。
總結(jié)
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
本文主要介紹了python 提高性能的方法。具有很好的參考價值，下面跟著小編一起來看下吧2017-03-03
圖像的輪廓檢測不論是機器視覺還是其他方面都有較大作用,下面這篇文章主要給大家介紹了關于如何使用Python?OpenCV提取物體輪廓的相關資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下2022-05-05
這篇文章主要給大家分享了關于Flask 上下文詳細介紹，我們可以把上下文理解為當前環(huán)境的快照，是一個用來保存狀態(tài)的對象。在代碼執(zhí)行的某個時刻，根據(jù)上下文的代碼邏輯，可以決定在當前時刻下使用到的環(huán)境變量等。下面一起進入文章了解詳情內(nèi)容,需要的朋友也可以參考一下2021-11-11
在本文里小編給大家分享了關于python字典如何改變value值的相關知識點以及實例代碼，需要的朋友們學習下。2019-06-06
今天小編就為大家分享一篇Python: 傳遞列表副本方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
python 正則表達式貪婪模式與非貪婪模式原理、用法實例分析
這篇文章主要介紹了python 正則表達式貪婪模式與非貪婪模式原理、用法,結(jié)合實例形式詳細分析了python 正則表達式貪婪模式與非貪婪模式的功能、原理、用法及相關操作注意事項,需要的朋友可以參考下2019-10-10
大家好，本篇文章主要講的是python用turtle庫繪畫圣誕樹，感興趣的同學趕快來看一看吧，對你有幫助的話記得收藏一下，方便下次瀏覽2021-12-12
在Heroku云平臺上部署Python的Django框架的教程
這篇文章主要介紹了在Heroku云平臺上部署Python的Django框架的教程,Heroku云平臺使用了Git版本控制系統(tǒng),所以本教程主要提供了配置所需要的Git腳本,需要的朋友可以參考下2015-04-04
Python OpenCV 使用滑動條來調(diào)整函數(shù)參數(shù)的方法
這篇文章主要介紹了Python OpenCV 使用滑動條來調(diào)整函數(shù)參數(shù)的方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-07-07
這篇文章主要介紹了Python中特殊函數(shù)，主要介紹一下四個函數(shù)：1 過濾函數(shù)filter 2 映射和歸并函數(shù)map/reduce 3 裝飾器@ 4 匿名函數(shù)lamda,需要的朋友可以參考下2015-07-07