最近又被裝環(huán)境折磨，僅根據(jù)實際情況記錄糟心經(jīng)歷。僅作記錄。

網(wǎng)上很多pytorch GPU版本安裝教程里一般步驟都說是：裝顯卡驅(qū)動、裝cuda、裝cudnn、最后安裝pytorch GPU環(huán)境。

但也有教程說可以不單獨安裝cuda、cudnn，直接安裝pytorch GPU環(huán)境。

于是，我打算在不安裝cuda、cudnn的情況下（主要是嫌安裝太麻煩），直接安裝pytorch GPU環(huán)境（最后安裝好是pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.6）。

• 機器情況：新機器，自帶的有nvidia-smi顯卡驅(qū)動。
• 機器：Ubuntu22.04 + 英偉達顯卡（GeForce RTX 4090）+ Anaconda3。
• 目標：安裝gpu版本的PyTorch1.12.1（torch1.12.1/cu116 + torchvision0.13.1/cu116）

一、電腦相關(guān)信息

1. 電腦顯卡環(huán)境

• 顯卡：Nvidia GeForce RTX 4090
• 電腦的顯卡驅(qū)動版本：525.105.17，CUDA Version：12.0

（CUDA Version12.0表示此顯卡驅(qū)動最高支持cu12.0。）

二、安裝Pytorch1.12.1/cu116（GPU版本）

1. 準備：新建虛擬環(huán)境

1）首先安裝Anaconda。（很簡單，自行搜索）

2）其次在用Anaconda建一個虛擬環(huán)境。

conda create -n torch1-12-1 python=3.10
#（torch1-12-1 表示自己起的環(huán)境的名字。）

3）激活虛擬環(huán)境。（后續(xù)的操作都是在該虛擬環(huán)境下進行的。）

conda activate torch1-12-1

2. 方式一

用pip在線安裝torch1.12.1+cu116、torchvision0.13.1+cu116：（在pytorch官網(wǎng)的歷史版本里找安裝命令）

1）在PyTorch官網(wǎng)歷史版本鏈接（ Previous PyTorch Versions | PyTorch）里面，找到使用pip安裝pytorch1.12.1/cu116、torchvision==0.13.1+cu116的安裝命令，進行在線安裝。

安裝命令：

# CUDA 11.6
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116

安裝很快，很快就安裝好了。測試torch的版本以及torch.cuda.is_available()都沒有問題。

但是，在用這個pytorch環(huán)境運行代碼時，總是報關(guān)于cuda的錯（此代碼在別的老機器上沒有問題）。

最后感覺可能是因為pip命令安裝的時候沒有順帶安裝cuda，再加上自己電腦自己沒有安裝cuda，所以感覺是環(huán)境沒有安裝好然后運行代碼就一直報錯。

于是選擇用下面的方式二重新安裝pytorch環(huán)境。

3. 方式二

用conda在線安裝pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.6：（在pytorch官網(wǎng)的歷史版本里找安裝命令）

最后選擇用conda命令安裝pytorch GPU版本。因為conda命令里面帶的有cudatoolkit。

1）在PyTorch官網(wǎng)歷史版本鏈接（ Previous PyTorch Versions | PyTorch）里面，找到使用conda安裝pytorch1.12.1/cu116、torchvision==0.13.1+cu116的安裝命令，進行在線安裝。

安裝命令：

# CUDA 11.6
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge

但是用conda在線安裝pytorch環(huán)境的時候，下載包的時候老是斷掉，下了一點就斷了報錯了。來回重新運行conda安裝的命令很多次，一直沒有下載安裝成功。

最后配了一下conda的鏡像源（照著這個鏈接里面配的anaconda | 鏡像站使用幫助 | 清華大學(xué)開源軟件鏡像站 | Tsinghua Open Source Mirror），不知道是因為這個原因還是突然網(wǎng)速變快了，突然一下子下載得賊快，幾分鐘就安裝好了。真的是有點玄學(xué)。。。。

如下圖：（下載安裝得很快）

測試：

使用下面的命令測試，可以看到torch和torchvision對應(yīng)的版本。

(base) ubuntu@ubuntu:~$ conda activate torch1-12-1
(torch1-12-1) ubuntu@ubuntu:~$ python
Python 3.10.11 (main, Apr 20 2023, 19:02:41) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>> torch.__version__
'1.12.1'
>>> torchvision.__version__
'0.13.1'
>>> torch.cuda.is_available()
True
>>> torch.version.cuda
'11.6'
>>> torch.backends.cudnn.version()
8302
>>> exit()
(torch1-12-1) ubuntu@ubuntu:~$ 

PyTorch1.12.1 GPU版本（pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6）就安裝好了~ 可以愉快使用了~

總結(jié)

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

最新評論