1.Ubuntu16.04:在anaconda下安裝pytorch-gpu

轉(zhuǎn)自：Ubuntu16.04:在anaconda下安裝pytorch-gpu_莫等閑996的博客-CSDN博客

1 創(chuàng)建虛擬環(huán)境并進(jìn)入

conda create -n pytorch-gpu python=3.6
conda activate pytorch-gpu

2 下載對(duì)應(yīng)的安裝包和配件

方法一(推薦)：

conda install pytorch-gpu

conda install mkl

方法二：

到清華鏡像下載對(duì)應(yīng)版本的pytorch包https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

到下載好的目錄下安裝：

cd ~/Downloads/
conda install --offline pytorch-1.0.1-py3.6_cuda8.0.61_cudnn7.1.2_0.tar.bz2

conda install mkl

3 檢驗(yàn)

python

import torch

print(torch.cuda.is_available())
————————————————
版權(quán)聲明：本文為CSDN博主「莫等閑996」的原創(chuàng)文章，遵循CC 4.0 BY-SA版權(quán)協(xié)議，轉(zhuǎn)載請(qǐng)附上原文出處鏈接及本聲明。
原文鏈接：https://blog.csdn.net/qq_41874879/article/details/107984378

------------------------------------------------------------------------------------

2.pytorch gpu不可用 原因核查

轉(zhuǎn)自：pytorch gpu不可用 原因核查_(kāi)yuuzhao的博客-CSDN博客_pytorch無(wú)法調(diào)用gpu

首先檢查pytorch cuda 顯卡驅(qū)動(dòng) 之間的對(duì)應(yīng)關(guān)系

查看pytorch對(duì)應(yīng)的cuda版本

打開(kāi)終端輸入：python

輸入命令：import torch

再輸入：print(torch.version.cuda)

查看pytorch 版本
print(torch.__version__)?
?

查看 cuda 是否安裝成功

顯示 torch 安裝成功，但 pytorch 對(duì)應(yīng)的cuda版本未安裝成功

查看cuda 版本

cat /usr/local/cuda/version.txt
1

查看是否能使用GPU

import torch
print(torch.cuda.is_available())
1
2

torch.cuda.is_available()
cuda是否可用；

torch.cuda.device_count()
返回gpu數(shù)量；

torch.cuda.get_device_name(0)
返回gpu名字，設(shè)備索引默認(rèn)從0開(kāi)始；

torch.cuda.current_device()
返回當(dāng)前設(shè)備索引；
1
2
3
4
5
6
7
8
9
10
11
12
CUDA的版本一定要和CUDNN的版本對(duì)應(yīng)，必須是CUDNN支持的版本！

查看Cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
1

Cudnn版本為： 7.6.5

查看顯卡驅(qū)動(dòng)版本

查看GPU型號(hào)

lspci | grep -i nvidia
1
查看NVIDIA驅(qū)動(dòng)版本

sudo dpkg --list | grep nvidia-*
1
或者

cat /proc/driver/nvidia/version
1

CUDA 與 顯卡驅(qū)動(dòng)的對(duì)應(yīng)關(guān)系核對(duì)圖

顯卡驅(qū)動(dòng)與cuda 沒(méi)問(wèn)題！

CUDA 與Cudnn 版本核對(duì)圖

???????

?

https://developer.nvidia.com/rdp/cudnn-archive

目前官網(wǎng)上沒(méi)有提供cuda10.2對(duì)應(yīng)的cudnn版本，不過(guò)本系統(tǒng)下載的是7.6.5,應(yīng)該是匹配了cuda10.2版本的

cuda 與cudnn 版本一致。

都檢查一遍之后 ，發(fā)現(xiàn) 是裝錯(cuò)pytorch 版本了， 裝的是 CPU版本的

# 進(jìn)入虛擬環(huán)境
conda activite pytorch-env

#查看torchvision 版本
conda list torchvision

# 進(jìn)入python 環(huán)境
python

#
import torch
torch.__version__
1
2
3
4
5
6
7
8
9
10
11
12

所以！！

重新 安裝 gpu 版本的 pytorch 即可！！

附件：

如何查看 pytorch - cuda - cudnn - python - nvidia 之間的對(duì)應(yīng)關(guān)系：

https://blog.csdn.net/yinxian9019/article/details/106782315

zhaoyuyu_nudt
關(guān)注

17

2

60

專欄目錄
pytorch gpu版
05-01
win10 pytorch 安裝包（百度網(wǎng)盤鏈接） 內(nèi)含： torch-1.7.1+cu110-cp38-cp38-win_amd64.whl torchaudio-0.7.2-cp38-none-win_amd64.whl torchvision-0.8.2+cu110-cp38-cp38-win_amd64.whl
windows下 anaconda + cuda + pytorch 安裝后gpu不可用問(wèn)題
星月夜話
?6577
本記錄使用的軟件版本截圖： 一、關(guān)鍵問(wèn)題 1、cuda版本 cuda 使用 2、NVIDA控制面板查看本機(jī)顯卡驅(qū)動(dòng)版本 二、注意事項(xiàng) CUDA版本選擇 不能選擇None ，none表示不使用GPU ***** 由于開(kāi)始使用的是ancaonda安裝的pytorch , 導(dǎo)致 pycharm中 torch.cuda.is_available() ...
?

?
Defsign3r:。。各種版本都對(duì)應(yīng)得上，沒(méi)想到啊沒(méi)想到26 天前回復(fù)
————————————————
版權(quán)聲明：本文為CSDN博主「zhaoyuyu_nudt」的原創(chuàng)文章，遵循CC 4.0 BY-SA版權(quán)協(xié)議，轉(zhuǎn)載請(qǐng)附上原文出處鏈接及本聲明。
原文鏈接：https://blog.csdn.net/yinxian9019/article/details/106763892

總結(jié)

以上是生活随笔為你收集整理的cuda gpu相关汇总的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。