CUDA安装调试

cuDNN
E:\CUDA\CUDA Documentation\extras\demo_suite
执行bandwidthTest.exe
CUDA安装调试_第1张图片
执行deviceQuery.exe
CUDA安装调试_第2张图片

稳定版本
CUDA安装调试_第3张图片

pytorch支持的11.3版本,之前下载的是11.4
CUDA安装调试_第4张图片

在YOLOv5的View-terminal中先import torch,
再输入命令:
print(torch.cuda.is_available())
CUDA安装调试_第5张图片

CUDA安装调试_第6张图片

先删除torch还有torchvision,因为好像这俩是对应cpu版本的
CUDA安装调试_第7张图片

然后重新下载
pip install torch-1.12.0+cu113-cp38-cp38-win_amd64.whl
pip install torchvision-0.13.0+cu113-cp38-cp38-win_amd64.whl
(上面这两条命令不行)
在Base环境中安装对应cuda11.3的pytorch
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
CUDA安装调试_第8张图片
(以上这个图是在删除pytorch之前下载的,所以很快)

以上是【在线下载】方式,但是删除后再次西在cuda版本的,太慢了
离线下载的方式前面要加绝对路径
即刚才出错的原因在于【没有添加绝对路径】,应该这么写:
pip install D:\下载的安装包\Python\torch-1.12.0+cu113-cp38-cp38-win_amd64.whl
pip install D:\下载的安装包\Python\torchvision-0.13.0+cu113-cp38-cp38-win_amd64.whl
CUDA安装调试_第9张图片

很快啊,啪的一下就ok了
CUDA安装调试_第10张图片

安装成功后的测试
print(torch.version)
CUDA安装调试_第11张图片

CUDA安装调试_第12张图片

注意\cuda\lib\x64\文件夹下的cudnn.lib复制到CUDA安装目录\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64\文件夹中
即cuDNN中的lib内容放到cuda->lib->x64下面

你可能感兴趣的:(python学习,windows)