CUDA与Pytorch安装指南
本文档详细介绍了如何在计算机上安装CUDA和cuDNN，以及如何为Python环境配置Pytorch。首先，从Pytorch官网选择合适的CUDA版本进行下载，然后通过conda创建并激活虚拟环境，接着使用pip3安装指定版本的Pytorch、torchvision和torchaudio。最后，验证安装是否成功，只需在Python环境中导入torch模块。
首先安装配置好CUDA和cuDNN，参考我之前的一篇笔记。
1、Pytorch下载
在Pytorch官网 根据对应版本选择下载：
复制command。
打开cmd，创建虚拟环境：
conda create -n Pytorch python==3.7然后激活环境：
conda activate Pytorch然后安装Pytorch，安装时间会有点久~
pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio===0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
检验是否安装成功：
输入 python ，import torch 即可 
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
2892
13万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
