pytorch安裝教程

weixin_46165555發表於2020-10-29

1.Anaconda 下載和安裝

下載Anaconda

找到連結: https://www.anaconda.com/products/individual
選擇好適合的版本
在這裡插入圖片描述

安裝Anaconda

接下來就是安裝了,安裝好之後,檢查一下是否安裝成功

開啟Aanconda Prompt
在這裡插入圖片描述
點選之後進入

在這裡插入圖片描述
如果可以在彈出的黑色命令列中看到base就意味著安裝成功了

2.顯示卡配置

開啟工作管理員
在這裡插入圖片描述
也許,你之後會遇到不同的專案,需要使用到不同版本的環境。比如這個專案要用到 pytorch 0.4,另一個專案要用到 pytorch 1.0,如果你解除安裝了0.4版本,安裝了1.0版本。那麼下一次,你再碰到0.4版本,你就需要解除安裝1.0版本,安裝0.4版本。很折騰。

Anaconda 整合的 conda 包就能夠解決這個問題。它可以創造出兩個屋子,相互隔離。一個屋子放 0.4 版本,一個屋子放 1.0 版本。你需要哪個版本,就進哪個屋子工作。

我們首先使用 conda 指令建立一個屋子,叫做 pytorch。
輸入命令列 conda create -n pytorch python=3.8
之後,彈出提示,輸入 y,即可安裝。
在這裡插入圖片描述
安裝成功之後,我們要在 pytorch 環境中,安裝 PyTorch,(有點繞),使用如下指令,進入 pytorch 環境。
在這裡插入圖片描述

你可以看到左邊的 base 變成了 pytorch,代表成功進入 pytorch 環境。

3.pytorch安裝

開啟官網並選擇合適的版本

在這裡插入圖片描述
往下拉會發現
在這裡插入圖片描述
根據自身情況選擇合適的版本
第一個推薦選擇Stable(1.7.0)
第二個選擇作業系統
第三個主要涉及安裝方式一般是Conda和Pip,Windows推薦選擇Conda,Linux推薦選擇Pip
第四個根據安裝的python的版本選擇
第五個,檢視GUP的型號,進行選擇

安裝pytorc

在這裡插入圖片描述
在命令列中輸入此命令,回車,之後出現很多的包,問你是否要進行安裝,果斷輸入y,之後就是漫長的等待。(我差不多是等了三個小時)
突然安裝到84%報錯了,我當時就絕望
在這裡插入圖片描述
根據報錯的資訊,我刪除了幾個檔案,一次又一次的嘗試安裝,歷經幾個小時,終於成功了。

4.驗證是否安裝成功

(1)在命令列左邊為 pytorch 環境中,輸入 python
(2)之後,輸入 import torch,如果沒有報錯,意味著 PyTorch 已經順利安裝了
(3)接下來,輸入 torch.cuda.is_available,如果是 True,意味著你可以使用 GPU,如果是 False,意味著只能使用CPU。

在這裡插入圖片描述

相關文章