PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU

一、Anaconda安装

anaconda下载地址清华镜像源下载

PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第1张图片

安装的时候跳过vscode

PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第2张图片

 添加路径

 二、安装显卡驱动

驱动下载地址

PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第3张图片

PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第4张图片

下载之后检查gpu运行情况

ctrl+alt+delete快捷键进入任务管理器

 三、安装CUDA

cuda下载地址

 PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第5张图片

PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第6张图片

 四、下载CuDNN

cudnn下载地址

 PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第7张图片

PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第8张图片 PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第9张图片

将下载好的cudnn这三个文件夹复制粘贴到 CUDA路径下 

 

添加路径

 五、验证cuda安装成功

验证cuda是否安装成功,首先win+R启动cmd,进入到CUDA安装目录下的 ...\extras\demo_suite,然后分别运行bandwidthTest.exe和deviceQuery.exe,应该得到图28,返回Result=PASS表示cuda安装成功。

PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第10张图片

 

 PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第11张图片

 六、下载pytorch

 

PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第12张图片

PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第13张图片 pytorch下载命令下对应的cuda

 

出现这个报错的话 可能是网络代理冲突 关掉科学上网

PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第14张图片

 

 PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第15张图片

 PyTorch GPU环境配置 win11+NVIDIA GeForce RTX3060 Laptop GPU_第16张图片

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