当前位置: 代码网 > it编程>前端脚本>Python > pytorch通过miniconda安装的实现示例

pytorch通过miniconda安装的实现示例

2025年06月03日 Python 我要评论
最近跟着李沐老师在学习动手深度学习,所以安装一下pytorch环境,之前自己通过conda安装过一次,新建了虚拟环境,踩了很多坑,这次想直接在原来的基础上继续用,结果d2l这个包一直装不上,索性直接全

最近跟着李沐老师在学习动手深度学习,所以安装一下pytorch环境,之前自己通过conda安装过一次,新建了虚拟环境,踩了很多坑,这次想直接在原来的基础上继续用,结果d2l这个包一直装不上,索性直接全删了,重新再来一遍,刚好c盘被这个快撑爆了。

1. 确认有nvidia gpu

桌面右键查看更多选项—nvidia控制面板,可以看到我这个是垃圾的mx 250

2、先给显卡更新一下驱动

这样支持的cuda版本会多一点,

访问nvidia官网:人工智能计算领域的领导者 | nvidia,更新驱动,点击驱动程序

找到自己的显卡对应的驱动并下载: 

下载完之后,打开文件,一直点击下一步,更新完驱动。

3、查看现在支持的cuda版本

桌面右键查看更多选项—nvidia控制面板

可以看到现在支持的cuda版本最高为12.2.1

4、安装cuda

访问cuda下载:https://developer.nvidia.com/cuda-downloads

根据自己系统选择,选择本地下载

下载完成后,双击文件,会让你选择提取文件的文件夹,我选择在d盘新建了一个文件夹,安装完成之后,这个文件夹会被自动删除。

接下来,为了防止后面出问题,我就选择了精简安装(c盘战士,哭

接下来等待即可,最后安装完成。

查看是否安装是否成功:cmd输入nvidia-smi,注意高版本的cuda这条指令没有空格,李沐老师的安装过程用的nvidia -smi,我试了半天都不对。

好像有种说法是这里是指最高支持的cuda版本,可以再输入指令nvcc -v试一下

5、下载miniconda

选择需要的python版本的miniconda ,如果没有安装过python的,可以随便选,我选的python3.9版本,但是这个版本就算跟你自己的python不一样也没关系,反正后面选择python解释器的时候选这个就行。

 下载好之后,安装。

我选的安装位置是d盘,因为c盘不够用了,如果c盘充足,最好放在盘,省的后面出现啥问题,记住这个安装路径,后面需要用到。 

注意,安装路径不能有中文,其余都是默认选项

安装完之后,找到anaconda powershell prompt(miniconda)

查看我们的python环境,确认是3.9:

6、下载pytorch

在这个网站,或者直接搜pytorch

下拉找到下载:

windows—pip安装—python—选择cuda12.1(我的cuda版本是12.2,可以向下兼容),根据自己情况选择:

复制下方pip指令进入刚才的miniconda的powershell中,防止下载过慢,可以在pip最后加一个中科大的镜像源,-i https://pypi.mirrors.ustc.edu.cn/simple

下载完成之后,看看torch能不能用,使用简单的张量运算来检查一下:

可以看到torch可以用,但是!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

注意:检查一下torch的版本,防止安装成cpu版本(悲),torch.__version__

明明选的gpu版本,为啥安装好了变成cpu版本了呢???解决方案可以看这个文章:

【精选】conda安装gpu版pytorch,结果却是cpu版本[找到问题根源,从容解决]

我的解决方案是先切换conda安装试一下,

在anaconda的powershell里试一下,如果出现提示,所有的包已经存在,那就换下一个方法,如果直接重新安装torch,等待安装完再看看torch的版本。

上面的方法不行,就只能使用使用本地安装,去conda官网:https://anaconda.org/,找到我要的pytorch安装包:

下载后把这个bz2文件放到:d:\miniconda\pkgs这个文件夹里面,就是刚在安装miniconda的哪个路径,然后在anaconda的powershell里面,先切盘到d:\miniconda\pkgs

然后,用conda命令进行本地安装:

conda install --use-local pytorch-2.1.0-py3.9_cuda12.1_cudnn8_0.tar.bz2后面这一串就是文件名

现在看看torch版本,后面不带cpu了,再试试能不能调用cuda,然后我的问题就解决了。

芜湖,起飞!

7.vscode运行

最后,我一般用vscode,然后在vscode里面,先选择python语言

然后在选择解释器,就是用我们刚才在miniconda里面安装的python解释器,因为我的miniconda的安装路径是在d盘,所以找到d盘里的python程序,最后地址是d:/miniconda/python.exe,刚才的torch也在这里。

选好之后我们测试一下最简单的线性模型:

win!!

总结

多看看安装教程,一定注意核对电脑支持的cdua版本+python版本+pytorch版本,版本兼容问题应该是新手遇到的最大的问题,其次,还有在conda中安装的方法,也可以尝试一下。

到此这篇关于pytorch通过miniconda安装的实现示例的文章就介绍到这了,更多相关pytorch通过miniconda安装内容请搜索代码网以前的文章或继续浏览下面的相关文章希望大家以后多多支持代码网!

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com