pytorch深度學習環境搭建踩坑實錄

python pytorch nvidia驅動 conda包管理器 cuda cuDNN

安裝Python

官網下載exe一路next,將以下路徑加入環境變量中的path
D:\Program Files\Python\Python37\Scripts
D:\Program Files\Python\Python37

安裝conda

簡介: Conda 是一個開源的軟件包管理系統和環境管理系統,用於安裝多個版本的軟件包及其依賴關係,並在它們之間輕鬆切換。
安裝:
官網下載miniconda:https://docs.conda.io/en/latest/miniconda.html
選擇Windows版本一路next,手動將安裝路徑中的scripts文件夾路徑加入環境變量中的path
cmd中輸入conda測試是否安裝成功
修改包管理鏡像爲國內源:
命令行:

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

也可以修改C:\users\username\中的.condarc配置文件,其中的channel改爲國內源
查看配置:conda config --show-sources
使用:
安裝包:

conda install package_name
conda install package_name1 package_name2

其他:https://blog.csdn.net/qq_37392932/article/details/81210470

安裝pytorch

因爲在代碼運行期間出現了這樣的報錯:The NVIDIA driver on your system is too old (found version 10000),說明我的NVIDIA driver和pytorch版本不匹配,所以我又把pytorh的版本降爲了1.0.0版本,重新進行安裝

在官網https://pytorch.org/找到命令行,利用conda進行安裝,可選擇與cuda匹配的先前的版本

如果conda不可用,可以選擇下載whl文件https://download.pytorch.org/whl/cu90/torch_stable.html
pip install 文件路徑 安裝

安裝過程如果出現報錯:ImportError: numpy.core.multiarray failed to import,可以使用pip install -U numpy進行安裝numpy模塊

torchvision

缺少torchvision,使用pip install torchvision命令行進行安裝,關於torchvision相關信息可以在官網https://pypi.org/project/torchvision/查看到。

安裝後代碼運行出現報錯: cannot import name ‘PILLOW_VERSION’ from ‘PIL’
原因: torchvision在運行時要調用PIL模塊的PILLOW_VERSION函數。但是PILLOW_VERSION在Pillow 7.0.0之後的版本被移除了,Pillow 7.0.0之後的版本使用__version__函數代替PILLOW_VERSION函數,可以將Pillow版本降爲6.1

pip uninstall pillow  //卸載7.0.0版本的pillow
pip install pillow==6.1  //安裝6.1版本

安裝cuda

發佈了51 篇原創文章 · 獲贊 8 · 訪問量 3萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章