windows10环境下安装pytorch
创始人
2025-05-28 02:48:42

安装Anaconda

Anaconda的安装按照下载安装的方式即可。
anaconda下载地址
anaconda安装教程

Nvidia显卡驱动程序安装

使用驱动管理软件下载显卡驱动并安装或者到显卡官网下载。
查看显卡的CUDA版本,如下图所示:
显卡信息
可以看到显卡驱动的版本是517.00。
显卡的CUDA版本是11.7。

CUDA安装

下载地址
CUDA的安装版本要低于电脑显卡的版本。我的电脑的CUDA版本是11.7,安装11.7以下的CUDA版本即可。在这里选择的是11.6。

cuDNN安装

下载地址
选择对应的版本安装即可。cuDNN的版本要对应CUDA的版本。

Pytorch安装

下载地址
pytorch尽量安装比自己CUDA版本低的版本。在这里安装的是pytorch11.3版本。在这里安装11.6版本的pytorch时一直提示找不到包,所以安装了11.3。

验证环境安装

显示如图则成功。
true

相关内容

热门资讯

中导光电科创板IPO获受理 系... 《科创板日报》6月29日讯(记者 陈俊清)近日,上交所官网显示,中导光电科创板IPO审核状态更新为已...
存储三巨头三星、海力士、美光被... 席卷全球的内存价格飙升,正演变为一场法律风暴。三星、SK海力士与美光这三家掌控全球DRAM市场的芯片...
台积电携手华邦供应DRAM晶圆... 全球记忆体供应持续吃紧,台积电正将本土厂商纳入其核心AI芯片供应链,以降低对国际三大记忆体巨头的依赖...
“玻璃桥”冲击到底有多大?华尔... 康宁GlassBridge“玻璃桥”技术的正式亮相在AI光通信产业链引发高度关注,但来自华尔街的评估...
超4755万亿韩元!韩国押注争... 韩国正把未来二三十年的国运押到AI上,这也将使得全球AI基建竞赛进入更为激烈的新阶段。6月29日,韩...