Windows下安裝Torch+cuda(Anaconda、Pycharm、NVIDIA驅動、Pytorch)、Jupyter
1.安裝Anaconda:
直接下載免費的 官方Anaconda安裝,安裝過程中建議自行配置安裝目錄,并確認添加運行環(huán)境到Path中,安裝后通過cmd命令行輸入conda -V 查看是否安裝以及版本號。
2.安裝Pycharm
社區(qū)版免費安裝,個人學習安裝 社區(qū)版 即可,社區(qū)版基本能滿足個人學習需求。
若安裝 專業(yè)版 需購買正版或自行破解,可尋找破解教程進行破解。
推薦Pycharm安裝兩個好用的插件:
- 代碼補全提示插件:Full Line Code Completion
- 主題插件:Matrial Theme UI
3.安裝NVIDIA驅動程序
在NVIDIA官網下載驅動程序并安裝,請確保自己下載的NVIDIA驅動版本和本機版本能適配。(請確保自己的設備支持GPU訓練)
安裝教程可參考:https://helpx.adobe.com/cn/premiere-pro/kb/drivers-video-win-nvidia.html
4.創(chuàng)建虛擬環(huán)境:
從開始菜單欄里面找到Anaconda Prompt并打開,然后創(chuàng)建虛擬環(huán)境:
conda create -n env_name python=x.x.x
- env_name:自定義的虛擬環(huán)境名稱
- python=x.x.x:虛擬環(huán)境對應的python版本
具體樣例:
conda create -n torch1 python=3.7.2
conda create -n torch2 python=3.8
5.安裝gpu版本的torch
首先進入虛擬環(huán)境:activate env_name
安裝torch+gpu版本:
注意:cuda版本要和剛剛下載的驅動版本相同或略小
Python版本和Torch版本以及GPU版本都有版本適配關系,本人通過如下命令成功安裝了torch1.13版本和torch2.0版本的安裝:
python3.7.2+:
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
python3.8+:
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
本人多次嘗試了上述命令,若網絡正常的話可以完全順利安裝,不會報錯。若安裝失敗或者卡住不動可嘗試重新輸入命令安裝。若多次嘗試仍無法正常安裝,安裝失敗的原因有很多,此處難以一一列出解答,建議根據(jù)報錯提示尋找對應的解決方案。
若希望安裝其他版本,可參考 Pytorch官方下載 提示進行下載。
注意!:先明確問題,再嘗試解決,不要在沒有明確自己的問題之前就跟著其他教程進行一些你看不懂的操作。
6.檢查torch是否安裝以及是否可用gpu
進入虛擬環(huán)境中的python shell中,查看torch版本并 確認是否可用gpu
進入虛擬環(huán)境:activate env_name
進入python shell:python
查看torch版本并 確認是否可用gpu:
import torch
print(torch.__version__) # x.xx.x+cuxxx
print(torch.cuda.is_available()) # True
具體樣例:
沒有+cuxx(僅安裝了cpu版本) 或者 false說明gpu版本的torch安裝有問題。
安裝失敗的原因有很多,此處難以一一列出解答,建議根據(jù)報錯提示尋找對應的解決方案。
注意!:先明確問題,再嘗試解決
7.在Pycharm中添加運行環(huán)境
在Pytorch的設置里面找到Project->Python Interpret,Add Interpret 添加.exe運行環(huán)境。
8.安裝其他的包
自行下面安裝其他的包, 例如 transformer,tqdm,jupyter等
pip install package_name
安裝jupyter后,可通過輸入 jupyter notebook 進入jupyter。
9.Jupyter配置
修改jupyter打開后的默認文件夾(根據(jù)自身實際情況設置)
配置文件目錄:C:\Users\username.jupyter\jupyter_notebook_config.py
修改內容:c.NotebookApp.notebook_dir=‘E:\username\xxx\Jupyter’
后面’’中的值為自己希望jupyter notebook打開后的默認文件夾的絕對地址
jupyter主題配置:
jupyter默認的主題和字體不是很美觀,可以參考如下方式修改一下:
可參考:https://www.zhihu.com/question/40012144
步驟如下:
-
從cmd命令行中找到本地jupyter notebook的插件路徑:
- jupyter --data-dir
-
進入剛剛的目錄,并創(chuàng)建文件夾:nbextensions\jupyter_themes
-
下載 配置文件 ,并將該配置文件放入 jupyter_themes 中。
-
激活主題
- jupyter nbextension enable jupyter_themes/theme_selector
-
進入jupyter主頁,設置主題與字體:
- 設置主題:Cell -> Code Syntax Theme
- 設置字體:Cell -> Code Font
本人主題修改目錄:C:\Users\yuq\AppData\Roaming\jupyter\nbextensions\jupyter_themes
-
推薦主題:dracula
-
推薦字體:Roboto Mono
當anaconda中的某一個環(huán)境安裝了jupyter后,可以在這個jupyter notebook當中使用其他任意環(huán)境的內核,即在notebook中使用其他的環(huán)境來寫代碼。前提是需要在目標環(huán)境中安裝jupyter內核,安裝jupyter內核的步驟如下:
查看是否已安裝內核:python -m ipykernel --version
若出現(xiàn)版本信息說明已安裝。否則安裝內核:python -m pip install ipykernel
安裝內核后,可以為jupyter notebook添加內核:
python -m ipykernel install --user --name=kernelname --display-name showname
kernelname為創(chuàng)建的文件夾名,showname為在notebook中展示的內核名,可一樣。
添加后可查看 Jupyter notebook 內核列表:jupyter kernelspec list
刪除內核:jupyter kernelspec remove kernelname
若文章有誤或需要改進的可以私信我或者評論區(qū)指出,謝謝!文章來源:http://www.zghlxwxcb.cn/news/detail-860591.html
如果覺得本文章有用的話,可以點贊關注支持一下(#^ ∨ ^#) ,謝謝!文章來源地址http://www.zghlxwxcb.cn/news/detail-860591.html
到了這里,關于Windows下安裝Torch+cuda(Pytorch、Anaconda、Pycharm、NVIDIA驅動)、Jupyter的文章就介紹完了。如果您還想了解更多內容,請在右上角搜索TOY模板網以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網!