1. 程式人生 > 程式設計 >Anaconda+spyder+pycharm的pytorch配置詳解(GPU)

Anaconda+spyder+pycharm的pytorch配置詳解(GPU)

第一步 : 從清華大學開源軟體映象站下載Anaconda:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D

在這裡插入圖片描述

安裝過程中需要勾選如下圖

在這裡插入圖片描述

裝好後測試是否裝好,先配置環境變數(可能anaconda安裝好後自己就有了)

在這裡插入圖片描述

開啟CMD,輸入程式碼

conda list

在這裡插入圖片描述

回車出現包的資訊則說明安裝完成

在這裡插入圖片描述

開啟Anaconda Navigator(桌面沒有的話就點選左下角看最近新增)可以看到spyder已經下好了

在這裡插入圖片描述

第二步:下載CUDA(GPU)
注意:沒有NVIDA的顯示卡是不能使用CUDA的!!!!!!!!!

如果顯示卡不是N卡的話,就不能使用GPU加速的!!!!!
但親測之後就算沒有下載CUDA,只要pytorch包下載在spyder裡,是能使用CPU來加速的,進行一些簡單的學習是沒有問題的。

下載CUDA前,檢視自己電腦能安裝CUDA的最高版本,開啟NVIDA控制面板

在這裡插入圖片描述

在這裡插入圖片描述

可以看到我的CUDA能安裝11.1版本的(這不代表你的電腦就安裝這個版本,而是最高能接受這個版本!)

然後我們進入CUDA工具包安裝官網
https://developer.nvidia.com/cuda-toolkit-archive

在這裡插入圖片描述

可以看到有很多版本,但這個時候可以看一下pytorch官網
https://pytorch.org/get-started/locally/

在這裡插入圖片描述

這裡最高CUDA選擇10.2,但由於本人的電腦就是裝不了10.2,所以選擇了10.1 updata2

在這裡插入圖片描述

點進去是這個樣子的

在這裡插入圖片描述

然後右下角下載好後開始安裝,遇到下圖這個注意勾選。

在這裡插入圖片描述

安裝過程如果出錯的大概率是你的電腦沒有 Visual Studio,如果真出錯了可以下個VS2019試一試,我就是出錯在這。

安裝完成後測試是否安裝完成,開啟CMD輸入

nvcc -V

安裝完成的話就會出現下圖

在這裡插入圖片描述

可能會遇到nvcc不識別,我們就去找nvcc這個檔案(如圖所示路徑)

在這裡插入圖片描述

找到之後把這個路徑新增到環境變數

在這裡插入圖片描述

新增之後,再次進入cmd,輸入nvcc -V測試安裝情況

到此CUDA安裝完成

第三步

Pytorch安裝
開啟cmd,輸入下面兩行程式碼(使用清華源加速)

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

先不要關閉cmd,然後進入pytorch官網
https://pytorch.org/get-started/locally/

在這裡插入圖片描述

熟悉的介面,因為我安裝的CUDA是10.1,所以CUDA選擇了10.1。
注意這段程式碼

在這裡插入圖片描述

必須去掉-c pyorch
所以我們在剛才的cmd中輸入

conda install pytorch torchvision cudatoolkit=10.1

然後稍等一會,cmd就會開始安裝了,安裝過程中碰到Y/N就選Y就行了,因為我已經安裝過了,所以cmd裡就是這樣的

在這裡插入圖片描述

等下載完成之後,開啟cmd,輸入

ipython

然後再如下圖輸入

import torch
torch.cuda.is_available()

在這裡插入圖片描述

如果輸出了一個True,則pytorch安裝完成了。

第四步
環境配置與測試
開啟Anaconda Navigator

在這裡插入圖片描述

在兩個環境中找一找有沒有pytorch一類的包

在這裡插入圖片描述

在這裡插入圖片描述

沒有的話只能重新下一次pytorch了
有的話開啟spyder,可以在工具選項設定為簡體中文

在這裡插入圖片描述

然後輸入程式碼測試

import torch
print('Torch Version:',torch.__version__)
print('CUDA GPU check:',torch.cuda.is_available())

在這裡插入圖片描述

正常輸出的話,那麼pytorch就裝好了

如果是要用pycharm來執行呢,則新建一個專案,選擇現有直譯器

在這裡插入圖片描述

然後在新增直譯器,在conda環境下如圖

在這裡插入圖片描述

在這裡插入圖片描述

在這裡插入圖片描述

然後建立就好了,建立完成以後,建立一個.py檔案,輸入程式碼

import torch
print('Torch Version:',torch.cuda.is_available())

執行如下,則配置完成了

在這裡插入圖片描述

到此這篇關於Anaconda+spyder+pycharm的pytorch配置詳解(GPU)的文章就介紹到這了,更多相關Anaconda spyder pycharm pytorch配置 內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!