根据已有安装的cuda配置合适的pytorch环境

目前网络上根据电脑配置安装合适的深度学习环境的帖子已经很多了,但是现实中会出现很久之前已经安装了对应的cuda,但是现在忘记了当时安装的是什么版本。本文针对这一问题展开攻略。

1 cuda安装版本查询

我们在查询自己应该安装什么版本的cuda时,一般是:
管理员运行windows powershell,然后输入命令:

nvidia-smi

在这里插入图片描述

之后根据显示的cuda版本,选择一个不大于显示版本号的cuda版本进行安装。因此,我们可以知道:nvidia-smi显示出的cuda版本仅仅只是本电脑的显卡驱动所支持的最高版本cuda。原先电脑上已有的cuda版本要通过该命令查询:

nvcc --version

在这里插入图片描述

2 虚拟环境创建

安装新的pytorch环境前创建一个新的虚拟环境,这里使用conda创建

conda create -n py38
conda activate py38

3pytroch安装

pytorch官网
接下来,在上述链接中选择对于cuda版本的pytorch安装命令,然后在anaconda cmd中的对应虚拟环境粘贴运行即可。
这里建议使用pip。

  • 9
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Python中使用CUDA(Compute Unified Device Architecture)需要在PyCharm中进行一些设置。如果你的计算机已经安装CUDA,你可以按照以下步骤配置PyCharm: 1. **安装CUDA和cuDNN**: - 确保你已经从NVIDIA官方下载并安装CUDA和cuDNN(如果适用)。这两个组件对于深度学习库如PyTorch或TensorFlow来说是必要的。 - 安装完成后,更新系统环境变量,以便在命令行中能够访问CUDA工具。 2. **安装PyCUDA or PyTorch with CUDA support**: - PyCUDA是一个用于PythonCUDA编程接口,而PyTorch有一个集成的CUDA支持。你可以选择安装PyCUDA或者直接安装PyTorch(如果你不打算用PyTorch的其他功能,这通常更简单)。 3. **配置PyCharm**: - 打开PyCharm,点击“File” -> “Settings” (or "Preferences" on Mac)。 - 在左侧导航栏找到“Project: [your_project_name]”,然后选择“Project Interpreter”。 - 如果你选择了安装PyTorch,可能会自动检测到CUDA。如果没有,点击“+”图标,选择“Python Interpreter” -> “Add local”或“Add Python interpreter from another location”,然后指定Python解释器路径,这个路径通常包含CUDA的bin目录(如`/usr/local/cuda/bin`)。 - 在PyCharm的“Run”菜单下,点击“Edit Configurations”,然后创建一个新的Python配置。 - 在“Program”字段中,选择包含CUDA代码的脚本。 - 在“Debugger”选项卡中,可能需要配置“Hardware Acceleration”为“CUDA”或“None”,具体取决于你的需求。 5. **验证配置**: - 运行一个包含CUDA操作的测试脚本,看看是否能够识别并利用GPU进行计算。如果配置正确,应该能看到GPU使用情况。 相关问题: 1. 我应该如何检查我的系统是否正确安装CUDA和cuDNN? 2. 如何在PyCharm中查看和修改系统环境变量? 3. 如果我在配置过程中遇到错误,如何调试?
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值