Win11+RTX3060配置CUDA等深度学习环境(转)

Win11+RTX3060配置CUDA等深度学习环境义目录标题)

记录安装步骤方便自己回看,转载自博客https://blog.csdn.net/weixin_43760844/article/details/115612970

1、下载准备

相关的安装包比如Anaconda、CUDA、CUDNN、Pytorch、TensorFlow等都可以在python环境配置相关链接找到相应版本下载。这里所有的库都是下载到本地后再安装的。

1、torch相关版本对应
2、tensorflow相关版本对应

2、下载安装Anaconda

3、下载安装CUDA和cuDNN

3.1 cuda和cuDNN下载

1、先下载Visual studio才能安装CUDA
2、CUDA官网地址
3、cuDNN下载

3.2 cuda和cuDNN安装

cuDNN安装:
1、打开cuDNN压缩包文件,分别将三个文件中的子文件复制到CUDA对应目录下Win11+RTX3060配置CUDA等深度学习环境(转)_第1张图片
2、例如bin文件Win11+RTX3060配置CUDA等深度学习环境(转)_第2张图片
Win11+RTX3060配置CUDA等深度学习环境(转)_第3张图片
3、其中lib文件为例外,需要复制到x64文件夹中
Win11+RTX3060配置CUDA等深度学习环境(转)_第4张图片
4、重启

4、安装GPU版pytorch与TensorFlow

1、pytorch官网
2、tensorflow官网

4.1 下载

4.2 安装

5、总结

测试pytorch

python
import torch
torch.cuda.is_available()

返回True表示安装成功。

你可能感兴趣的:(深度学习,tensorflow,pytorch)