下载地址
官方下载：CUDA和CUDNN.
安装CUDA
安装之前，建议关掉360安全卫士
双击cuda_11.0.3_451.82_win10.exe文件
根据自己需要更改安装路径
将Visual Studio Integration的勾去掉
配置环境变量
安装CUDNN
将下载的CUDNN解压缩，如下图。
将将CUDNN文件夹里面的bin、include、lib文件直接复制到CUDA的安装目录，如下图为CUDA的安装位置，粘贴过来直接覆盖即可。
# CUDA的安装目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0等待复制完成，即可！
验证CUDA是否安装成功
打开cmd，输入如下命令，即可！
nvcc -V安装tesorflow-gpu2.4.1
查看对应版本
https://tensorflow.google.cn/install/source_windows#gpu
测试代码
import tensorflow as tfimport osos.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
查看对应版本
https://pytorch.org/get-started/previous-versions/
测试代码
import torchprint(torch.__version__)print(torch.cuda.is_available())安装paddlepaddle-gpu2.0.0
查看对应版本
https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/2.0/install/pip/windows-pip.html
测试代码
import paddlepaddle.utils.run_check()到此这篇关于Win10下安装CUDA11.0+CUDNN8.0+tensorflow-gpu2.4.1+pytorch1.7.0+paddlepaddle-gpu2.0.0的文章就介绍到这了,更多相关tensorflow pytorch CUDA 安装内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持！
最新文章
热门文章
猜你喜欢
