摘要：本文基于 PyTorch训练框架 电脑装了CUDA 但是python训练的时候显示的还是用了cpu进行训练 def print_device_info(): """打印设备信息""" print("PyTorch版本:", torch.__version__) print("CUDA是否可用:", 阅读全文

摘要：1、查看支持的CUDA版本 查看系统信息 这里CUDA12.3.107 是支持的最高版本的CUDA，可以向下兼容，且可以安装多个版本的CUDA，可以通过更改环境变量来更改为你需要用到的CUDA版本。 2、下载对应CUDA版本 CUDA下载地址：https://developer.nvidia.com 阅读全文

摘要：当我们本地Python开发下载Hugging Face模型的时候 下载的模型默认缓存是在用户盘底下的 一般是C盘，但是模型一般都比较大，这就导致C盘系统盘爆满，所以要修改默认缓存路径 点击WIN+R 输入control 如下图所示 打开 控制面板 > 系统和安全 > 系统 > 高级系统设置 如果在上 阅读全文