首先安装配置好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 即可 
本文档详细介绍了如何在计算机上安装CUDA和cuDNN,以及如何为Python环境配置Pytorch。首先,从Pytorch官网选择合适的CUDA版本进行下载,然后通过conda创建并激活虚拟环境,接着使用pip3安装指定版本的Pytorch、torchvision和torchaudio。最后,验证安装是否成功,只需在Python环境中导入torch模块。
1466

被折叠的 条评论
为什么被折叠?



