Windows下PyTorch深度学习环境配置(GPU)

一:下载Anaconda

(路径最好全英文)

(下载好后,可以创建其他虚拟环境,因为是自己学习,所以先不放步骤,有需要者可以参考B站up我是土堆的视频)

二:利用 conda 或者 pip 安装 PyTorch

1. 确定自己安装的CUDA版本与GPU关系

1.确定显卡型号

(如图右上角,我是1050ti)

  1. 确定显卡算力 6.1

    (更多CUDA和GPU间的算力关系可参考https://zhuanlan.zhihu.com/p/544337083?utm_id=0)
  2. 确定CUDA Runtime

    (可以看到算力6.1对应的CUDA SDK,我选用目前最新的11.8)
  3. 我们看自己的驱动 CUDA Driver Version -- 12.2
    (打开cmd,输入nvidia-smi 就可看到cuda的版本,建议更新到最新)
    在-
    https://www.nvidia.cn/Download/index.aspx?lang=cn进行安装

    (将其中的信息选择自己电脑对应的型号,之后进行安装)

2. 安装GPU版本 PyTorch

打开pytorch官网

(本人如上)

将最下面那行复制,打开Anaconda prompt的窗口复制进去进行下载

(这里有镜像和pip的方法)

补pip:就是把第三行的conda换成Pip,再复制最后一行代码进行下载

三:验证

cpp 复制代码
conda env list  //显示环境
conda list  //显示环境下的包
conda activate 环境名  //切换环境
conda uninstall pytorch torchvison torchaudio cudatoolkit  //卸载那四个包

四:安装PyCharm并进行配置



PS:有更多疑问可以参考B站我是土堆

PS:他的博客

相关推荐
Rocky Ding*33 分钟前
深入浅出GroundingPI技术解析:精确空间感知如何迁移到机器人动作
论文阅读·人工智能·深度学习·机器学习·aigc·ai-native·具身智能
如若1231 小时前
Windows SSH + Headscale/Tailscale 组网及免密登录完整配置教程
运维·windows·ubuntu·ssh·内网穿透
黑十1 小时前
PyTorch 报 “no CUDA GPUs are available“ 怎么办?3 步排查
服务器·网络·人工智能·pytorch·python
ZhengEnCi1 小时前
正样本几万、负样本个位数:极端类别不平衡下怎么训模型(全文可跑)
深度学习
布吉岛的石头2 小时前
Java 程序员第 50 阶段13:对齐税 alignment tax 是什么
人工智能·深度学习·transformer
ITHAOGE153 小时前
下载 | Win11 25H2 官方正式版ISO映像!(10月更新、消费者版/专业版、商业版/企业版、26200.9711)
windows·科技·微软·电脑
John_Hui3 小时前
Windows 10 安装版本:mysql-8.1.0-winx64.zip
数据库·windows·mysql
kingx33 小时前
Windows Server安全配置深度复盘:关键 Payload 与解题思路
windows·靶场·web·加固
技术人生0013 小时前
DirectX修复工具下载安装教程-Windows系统游戏dll修复工具下载安装修复教程
windows·游戏