浅浅踩了个小坑,按这个版本配置深度学习环境木有问题,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 | 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