4090D显卡-深度学习环境配置参数

最新推荐文章于 2024-07-06 19:34:24 发布

EstelleYuan

最新推荐文章于 2024-07-06 19:34:24 发布

阅读量599

点赞数 10

分类专栏：深度学习文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/jianai223/article/details/137600663

版权

深度学习专栏收录该内容

1 篇文章 0 订阅

订阅专栏

浅浅踩了个小坑，按这个版本配置深度学习环境木有问题，mark~~

1. conda、python、CUDA、CUDNN和Tensorflow版本配置

conda --version	23.7.4
python --version	3.10.13
nvcc --version	11.8.89

import tensorflow as tf tf.__version__	2.10.0

2. Pytorch版本配置

print(torch.__version__)	2.4.0
print(torch.version.cuda)	11.8
print(torch.backends.cudnn.version())	8700

Tips: 记得加入镜像地址

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
conda config --set show_channel_urls yes