rtx3070laptop+cuda11.3+cudnn8.2.2+pytorch1.10.2配置

shellydu

于 2022-09-06 15:25:21 发布

阅读量1.1k

点赞数

文章标签： pytorch 深度学习人工智能

本文链接：https://blog.csdn.net/shellydu/article/details/126725869

版权

该博客介绍了如何在拥有RTX3070显卡的笔记本上安装CUDA 11.3和cudnn8.2.2，以及后续如何通过Anaconda配置环境并安装PyTorch 1.10.2+cu113与torchvision 0.11.3+cu113。过程中强调了环境变量的配置和版本匹配的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

笔记本rtx3070，用nvidia软件查看cuda11.6，因11.6与pytorch不匹配，顾重新安装11.3版本

1.安装cuda11.3与cudnn8.2.2

参考

注意cudnn安装好后，要添加环境变量

选用python3.8

conda create -n pytorch-gpu-rtx3070 python=3.8
activate pytorch-gpu-rtx3070

在设置的环境内安装pytorch


pip install torch==1.10.2+cu113 torchvision==0.11.3+cu113 -f https://download.pytorch.org/whl/torch_stable.html

测试pytorch-cuda是否安装成功