目錄
- 一、anaconda 安裝
- 二、創建pytorch
- 1. 創建pytorch環境:
- 2. 激活環境
- 3. 下載安裝pytorch包
- 4. 檢查是否安裝成功
一、anaconda 安裝
具體的安裝說明可以參考我的另外一篇文章【環境搭建】Linux報錯bash: conda: command not found…
二、創建pytorch
1. 創建pytorch環境:
conda create -n dtc python=3.8
這里:
create 代表創建環境。
-n代表name, 也就是說我們創建的這個環境名字叫做dtc。可以根據自己的意愿設置環境名字。
python=3.8, 這里我們是指定python3.8版本,如果不指定版本的話會默認安裝最新版的python,同時也可以根據自己的需求設置python版本。
此時我們名字為dtc的環境只是一個軀殼,下面我們要進入這個環境中并安裝pytorch包。
2. 激活環境
conda activate dtc
可以看見,激活環境之后,我們由原先的base環境進入到被激活的dtc環境。
3. 下載安裝pytorch包
然后去pytorch官網找到適合自己電腦配置的pytorch版本進行安裝,在這里由于我是安裝在服務器終端,所以選擇了以下版本:
安裝命令在官網上選好配置后會顯示出來
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
將上述代碼直接復制到終端即可開始安裝pytorch包。
4. 檢查是否安裝成功
在pytorch環境下依次輸入下列語句:
python #檢查python版本
import torch
torch.cuda.is_available()
效果如下圖所示既表示安裝成功!
先寫到這里吧,到這里就基本安裝成功了,如果有什么報錯可以自行百度解決。
最后祝大家都可以順利安裝成功,成為一名優秀的算法工程師!