1.打开界面

按window键和R键打开运行,之后输入cmd按回车进入。

2.创建环境

进入后输入(以下所有输入复制粘贴即可)

conda create --name torch1.7 python=3.7

之后输入y表示确定,等待跳出C:/User后进行下一步。

3.激活环境

输入

conda activate torch1.7

会看到开头会有(torch1.7)

4.安装pytorch

输入

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

​

之后等待安装(文件比较大,可能等待时间很长),建议用自己手机热点进行下载(用无线可能比较慢或中途就停止本人亲测。)

5.检测安装是否可用

在pytorch1.7环境下,先输入python按回车,之后输入import torch按回车,再输入torch.cuda.is_available(),若显示True则表示安装成功。(如图)

 若不成功,则卸载后重新按操作进行安装。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐