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

安装

*
pytorch
查看Python版本
先查看CUDA版本
在线安装进入[官网](https://pytorch.org/)下载
离线下载安装
是否安装成功

pytorch

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

查看Python版本

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

; 先查看CUDA版本

nvidia-smi

【人工智能之人脸识别戴口罩】windows10上安装pytorch和使用【CUDA11.6版本】
如图我的是11.6版本的

在线安装进入 官网 下载

【人工智能之人脸识别戴口罩】windows10上安装pytorch和使用【CUDA11.6版本】
选择对应的版本和安装方式!
但是下载速度慢,所以可以离线安装

; 离线下载安装

pytorch与torchvision下载链接:
https://download.pytorch.org/whl/torch_stable.html

选择对应版本下载
例如:

【人工智能之人脸识别戴口罩】windows10上安装pytorch和使用【CUDA11.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

【人工智能之人脸识别戴口罩】windows10上安装pytorch和使用【CUDA11.6版本】
安装成功!

; 是否安装成功

运行代码看能否使用

import torch

【人工智能之人脸识别戴口罩】windows10上安装pytorch和使用【CUDA11.6版本】
运行成功🎉🎉🎉🎉!

Original: https://blog.csdn.net/qq_46556714/article/details/124780047
Author: 南蓬幽
Title: 【人工智能之人脸识别戴口罩】windows10上安装pytorch和使用【CUDA11.6版本】

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/709608/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球