当前位置: 首页 > 商场 >

焦点滚动:小白向:关于pytorch安装的一点经验分享~

来源:哔哩哔哩    时间:2023-01-19 20:03:31


(资料图片仅供参考)

完成微博舆情数据的收集后,下一步就是安装PyTorch 进行深度学习啦。安装过程中踩了一些坑,发出来和大家分享交流下~

1、为什么要用ANACONDA 配置深度学习的环境?

因为如图所示,用ANACONDA 可以初始化多个版本的环境,省去了多个项目文件需要配置多个虚拟环境,减少文件硬盘占用,方便管理虚拟空间。

2、安装过程的注意事项️

一定要确保显卡驱动的Cuda版本和安装的cuda版本是一致的。我这次安装从PyTorch 官网上就是报错版本号不对应,后面指定版本号后顺利安装,conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia 

PS:安装好后一定要检查是否能运行,如我最后一个图所示~希望大家都能装上自己的PyTorch!

上一篇:

下一篇:

X 关闭

Copyright   2015-2022 起点空净网版权所有   备案号:皖ICP备2022009963号-12   联系邮箱: 39 60 29 14 2@qq.com