快捷導(dǎo)航

Win10下安裝CUDA11.0+CUDNN8.0+tensorflow-gpu2.4.1+pytorch1.7.0+paddlepaddle-gpu2.0.0

更新時(shí)間：2021年03月23日 08:46:50 作者：FriendshipTang

這篇文章主要介紹了Win10下安裝CUDA11.0+CUDNN8.0+tensorflow-gpu2.4.1+pytorch1.7.0+paddlepaddle-gpu2.0.0，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

下載地址

官方下載：CUDA和CUDNN.

在這里插入圖片描述

安裝CUDA

安裝之前，建議關(guān)掉360安全衛(wèi)士

在這里插入圖片描述

雙擊cuda_11.0.3_451.82_win10.exe文件

在這里插入圖片描述

根據(jù)自己需要更改安裝路徑

在這里插入圖片描述

將Visual Studio Integration的勾去掉

在這里插入圖片描述

配置環(huán)境變量

在這里插入圖片描述

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\libnvvp;

在這里插入圖片描述

安裝CUDNN

在這里插入圖片描述

將下載的CUDNN解壓縮，如下圖。

在這里插入圖片描述

將將CUDNN文件夾里面的bin、include、lib文件直接復(fù)制到CUDA的安裝目錄，如下圖為CUDA的安裝位置，粘貼過(guò)來(lái)直接覆蓋即可。

# CUDA的安裝目錄
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0

在這里插入圖片描述

等待復(fù)制完成，即可！

驗(yàn)證CUDA是否安裝成功

打開(kāi)cmd，輸入如下命令，即可！

nvcc -V

在這里插入圖片描述

安裝tesorflow-gpu2.4.1

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

https://tensorflow.google.cn/install/source_windows#gpu

在這里插入圖片描述

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.4.1

在這里插入圖片描述

測(cè)試代碼

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
print(tf.__version__)
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print('GPU:', tf.test.is_gpu_available())

在這里插入圖片描述

安裝pytorch-gpu1.7.0

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

https://pytorch.org/get-started/previous-versions/

在這里插入圖片描述

pip install torch===1.7.0+cu110 torchvision===0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

在這里插入圖片描述

測(cè)試代碼

import torch
print(torch.__version__)
print(torch.cuda.is_available())

在這里插入圖片描述

安裝paddlepaddle-gpu2.0.0

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

https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/2.0/install/pip/windows-pip.html

在這里插入圖片描述

python -m pip install paddlepaddle-gpu==2.0.0.post110 -f https://paddlepaddle.org.cn/whl/stable.html

在這里插入圖片描述

測(cè)試代碼

import paddle
paddle.utils.run_check()

在這里插入圖片描述

到此這篇關(guān)于Win10下安裝CUDA11.0+CUDNN8.0+tensorflow-gpu2.4.1+pytorch1.7.0+paddlepaddle-gpu2.0.0的文章就介紹到這了,更多相關(guān)tensorflow pytorch CUDA 安裝內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: