Nvidia顯卡驅動安裝
安裝新的nvidia驅動,版本大于457.30(必須支持cuda11.3),不需要預先安裝cuda,
例如:
nvidia 457.30-desktop-win10-64bit-international-whql.exe
可以去英偉達官方網站下載,網址:Nvidia官網

我這里下載的版本是461.92.雙擊進行默認安裝即可,
安裝完成后可以打開cmd命令列 輸入Nvidia-smi 查看安裝的GPU CUDA版本

Anaconda虛擬環境
去Anaconda官網下載對應Python版本的anaconda 官網下載地址

下載好之后雙擊打開,進行安裝,

這里選擇ALL Users 將權限打開

安裝路徑盡量不要安裝在系統盤,因為后期安裝庫會遇到權限問題,出現conda安裝包報錯:The current user does not have write permissions to the target environment(當前用戶沒有寫入權限),導致第三方庫無法安裝,具體問題可以參考C:盤安裝報錯,盡量安裝到別的盤根目錄下,

這個地方第一個也要勾選,后面不用手動配置路徑,能夠自動配置,

點擊Install,等待安裝完成即可,
Cuda虛擬環境配置
添加conda源,加快安裝速度,在cmd中運行下面命令列:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
會在在c:/Users/用戶名/ 生成.condarc 檔案,
用記事本打開 .condarc 檔案,將里面所有內容用如下替換:
channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
ssl_verify: true
channel_priority: strict

然后 創建虛擬環境并激活,在cmd命令列中輸入:
conda create --name ef3090 python=3.7
conda activate ef3090
等待安裝下載即可,
Pytorch下載安裝
安裝pytorch,安裝cuda 11.3版本的pytorch
3090目前就11.3版本可以安裝cuda,前面的都不行,注意看安裝的時候,要安裝 pytorch/win-64::pytorch-1.10.1-py3.7_cuda11.3_cudnn8_0,如果是cpu后綴,就表示不支持cuda, -c pytorch表示從官網下載,速度慢,
cmd命令列輸入如下,進行安裝:
conda install pytorch torchvision torchaudio cudatoolkit=11.3
有時會出現conda安裝環境報錯:Solving environment: failed with initial frozen solve.
解決方法:
1.更新conda到最新版本:conda update -n base conda
2.再查一下conda版本:conda -V
3.第二次更新conda到最新版本:conda update -n base conda
4.更新完后再查一下conda版本:conda -V
5.執行:conda update --all 將所有庫升級為最新版本,時間比較慢,等待即可,
然后再次輸入conda install pytorch torchvision torchaudio cudatoolkit=11.3 即可
如果要在cuda中安裝第三方庫,在cmd輸入:
pip install pyyaml opencv-python tqdm matplotlib scikit-learn==0.22.1 -i https://pypi.mirrors.ustc.edu.cn/simple/
Pycharm配置Pytorch環境
打開File下的Settings

選擇Python Interpreter,再點擊右上角的小齒輪

選擇show all

點擊左上角的+號

選擇第二個Conda Evironment ,右側選擇第二項Existing environment ,Interpreter 選擇Anaconda下面的python.exe,不要選擇envs子目錄下的python.exe,因為呼叫的庫上面已經安裝在Anaconda-lib中了,選擇子目錄的python會發現第三方庫無法呼叫,Conda executable一般會自動配置選好,無需我們自己選擇,點擊OK即可,

結果
完成上述操作,Pytorch相關程式可以運行,寫此博客以防以后忘記,無其他用途,如有其他問題,歡迎評論區探討,
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/423340.html
標籤:AI
