巧妙的方法

直接在自己的环境终端下输入,我一般都是这样安装,很方便稳定。

pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110  -f https://download.pytorch.org/whl/torch_stable.html

复制过去自己改版本号,很有可能报错,报错的原因大概率是版本和cu后面的版本不匹配,这个时候去搜一下匹配的版本即可,也有可能是自己的GPU算力达不到安装的版本(这个我就遇到过,安装的版本跟GPU算力竟然也不匹配)

标准安装方法

1.创建环境

首先进入cmd,输入conda create --name torch python=3.8

##这里将新的环境名定义为torch,在python3.8的基础下安装

2.进入新环境

输入conda activate torch

##这时可以看到最左边会变成(torch),说明已经进入torch的环境

3.安装pytorch

进入pytorch官网PyTorch,点击上面的Get Started,然后根据自己的配置选择要安装的版本

##这里我选择的windows版本下的CPU,这能够满足一个普通研究生狗的需求,毕竟大的数据一般都放到服务器上跑。

这里复制最后一行的代码粘贴到cmd上

4.等待安装完成 可以用import torch调用看有没有安装成功。

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐