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

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐