Windows快速配置深度学习环境,调用显卡加速(cuda+pytorch+miniconda)

1.安装cuda

(需要nvidia显卡,集成显卡和AMD用不了cuda,直接跳到第二步)

cuda官网下载

一路默认安装

安装完成后到终端输入 nvidia-smi 验证是否安装成功

2. 安装miniconda

miniconda官网下载

选择自己电脑对应的安装包

一路默认安装

安装完后,进入miniconda的shell看一下安装的python版本是不是我们要的

3.安装GPU版本的PyTorch

(没有Nvidia GPU的在Computer Platform那里选择CPU)

pytorch官网

根据电脑选择好后,复制最后一串命令

(注意cuda版本要保持一致,比如你安装了cuda12.x ,这里就不能选择 pytorch-cuda11.x)

复制命令,去miniconda环境里安装即可

然后简单测试一下,能不能在GPU上创建一个张量并进行计算

大功告成

分享你的喜爱

留下评论

您的电子邮箱地址不会被公开。 必填项已用*标注