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

等待安装完成即可。

posted @   8bit布丁  阅读(102)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)
· spring官宣接入deepseek,真的太香了~
点击右上角即可分享
微信分享提示