栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

win11(amd)+cuda+cudnn+pytorch安装过程

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

win11(amd)+cuda+cudnn+pytorch安装过程

win11(amd)+cuda+cudnn+pytorch安装过程 Anaconda环境配置

“此电脑”右键属性,然后选择“高级系统设置”里的“环境变量”。

选择“Path”,并点击“编辑”。

添加三个环境变量:

注意:路径是你自己定义的安装路径。


在anaconda prompt运行下列命令:

 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/conda-forge/
 conda config --set show_channel_urls yes
CUDA安装 查看电脑Nvidia版本

win+s输入nvidia,打开进入控制面板


根据自己显卡版本选择相应的CUDA版本。

更新驱动

如果驱动不是最新版可以更新一下,根据自己电脑配置选择相应信息。地址:https://www.nvidia.cn/download/index.aspx?lang=cn



下一步后需要等待一段时间。


这部关闭即可。
后面需要登录一下账户,没有的话可以注册。
再次查看Nvidia版本,发现我的版本从11.4升级到11.7了。

下载CUDA

CUDA官网:https://developer.nvidia.com/cuda-toolkit
进入后选择“Download Now”,根据显卡版本选择相应的CUDA版本,注意不要选错系统和版本。

运行exe文件:

安装完成后,win+r,输入nvcc -V,有以下内容代表安装成功:

安装cuDNN

官网:
先进行登录,然后选择与CUDA匹配的版本。比如我的Nvidia是11.7,选择了for 11.x版本。

将Zip文件解压,将内部文件全部复制。

粘贴到如下位置。注意:如果你更改了安装位置,就粘贴到你的安装路径里。

环境变量配置

在“高级系统设置”的“环境变量”中“新建”。

根据自己安装的路径。
CUDA_LIB_PATH = %CUDA_PATH%libx64
CUDA_BIN_PATH = %CUDA_PATH%bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%binwin64
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%commonlibx64
验证

win+r,输入cd C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0extrasdemo_suite,再输入 .bandwidthTest.exe,得到如下结果,证明配置成功。

安装GPU版torch

在官网下载会非常慢,可以选择先将文件下载下来,再进行添加,pip/conda install 文件路径+文件名。清华镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/?C=M&O=A

注意文件命名格式,下图有错误示范。

分别添加pytorch、torchvision、cudatoolkit三个包后,再在命令窗口运行conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch注意:11.7是我电脑安装的nvidia的版本号。
等待添加成功后,分别输入下列代码:

python
import torch
torch.cuda.is_available()


这样就安装成功了!

本文是安装时的记录贴,如果有哪里不对的地方请指出,谢谢!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1029879.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号