Windows安装pytorch
新手第一次安装,记录自己安装经过。
1.安装anaconda
直接在anaconda官网上选择相应版本安装包安装即可。anaconda自带python,为避免和已有python冲突,可将已有版本删除。anaconda安装时记得一定要勾选上添加环境变量这一选项。成功安装后可在命令行中输入conda --version进行测试。
2.安装CUDA和cudnn
CUDA我之前装过了,直接就跳过了。不过安装时注意选择自定义安装,之后一路下一步就好。安装后的位置如图。
cudnn去官网下载,下载时要完成一个简单的注册才能下载。注意下载时选择和自己CUDA匹配的版本。下载下来是一个压缩包,只需要把每个文件夹的内容复制到CUDA中同名文件夹下即可完成安装,cudnn中单独的那个文件复制到上图文件夹中即可。
3.创建虚拟环境
打开anaconda prompt或者命令行,输入
1 | conda create -n pytorchlearn(虚拟环境名) python=3.9 |
完成创建虚拟环境,之后输入
1 | activate pytorchlearn |
即可激活环境。
4.安装torch和torchvision
在命令提示符中先pip安装wheel包,之后直接pip先安装torch,再安装torchvision。可能是使用了梯子的原因,安装速度不慢。
完成安装。
作者:y0hv2y
出处:https://www.cnblogs.com/y0hv2y/p/17028958.html
版权:本作品采用「署名-非商业性使用-相同方式共享 4.0 国际」许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具