【人工智能之人脸识别戴口罩】windows10上安装pytorch和使用【CUDA11.6版本】_pytorch-cuda=11.6

作者：凡人多烦事01 | 2024-06-08 20:47:56

踩

pytorch-cuda=11.6

学习需要安装pytorch，记录一下自己的安装过程！

在这里插入图片描述

nvidia-smi

在这里插入图片描述
如图我的是11.6版本的

在这里插入图片描述
选择对应的版本和安装方式！
但是下载速度慢，所以可以离线安装

选择对应版本下载
例如：
在这里插入图片描述

CUDA没有11.6的我选择11.5，Python版本3.8，Windows64位

下载完pytorch和torchvision的.whl文件之后，在下载的目录内使用下列命令即可：

pip install torch-1.11.0+cu115-cp38-cp38-win_amd64.whl
pip install torchaudio-0.11.0+cu115-cp38-cp38-win_amd64.whl

在这里插入图片描述
安装成功！

运行代码看能否使用

import torch
1

在这里插入图片描述
运行成功

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/凡人多烦事01/article/detail/691138