windows安装cuda/pytorch
安装cuda
笔者安装指定版本cuda的目的是安装gpu版本的pytorch,故先访问torch官网,查看需要的cuda版本。
访问:
https://pytorch.org/get-started/locally/
可以看到,在windows10下,使用conda安装,需要CUDA11.3
查看已经安装的cuda版本:
nvcc --version
如果命令不存在,或不符合要求,就要下载并安装对应版本的CUDA。
访问:
https://developer.nvidia.com/cuda-11.3.0-download-archive
根据你的平台,选择下载cuda版本。这里笔者下载windows版,等待下载完成,安装即可。
安装pytorch
首先创建虚拟环境:
conda create --name python39_torch1_9 python=3.9
注意创建时不能启用vpn,否则会报host_name的错误。
创建完毕后启用:
conda activate python39_torch1_9
从torch官网就能得到windows的安装命令:
https://pytorch.org/get-started/locally/ # 最新版本
https://pytorch.org/get-started/previous-versions/ # 历史版本
这里,我们选择历史版本:
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=11.3 -c pytorch -c conda-forge
等待安装完成即可。
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)
· spring官宣接入deepseek,真的太香了~