如何安装PyTorch+管理conda环境

一、检查电脑上是否有GPU
1.Windows
在Windows上,您可以使用设备管理器来检查是否有GPU:

右键点击开始菜单,选择“设备管理器”。
在设备管理器中,展开“显示适配器”部分。这里列出的设备会显示您的计算机上的所有图形硬件。如果您看到NVIDIA、AMD或Intel等品牌的条目,说明您有一个GPU。
在这里插入图片描述
打开NVDIA控制面板,查看GPU驱动版本
在这里插入图片描述
在这里插入图片描述

2.macOS
在macOS上,直接选择以下命令即可,直接用cpu,不必要用gpu加速
在这里插入图片描述

3.Linux
在Linux上,您可以使用命令行工具来检查GPU。打开终端,您可以使用以下命令之一:

对于NVIDIA GPU,使用nvidia-smi命令。
通用检查,使用lspci | grep VGA命令来查找所有的VGA兼容设备,这通常包括GPU。
请注意,如果您在使用这些命令时遇到问题,可能需要安装额外的驱动程序或工具。例如,nvidia-smi需要NVIDIA驱动程序已安装才能正常工作。

二、查找与本地GPU对应的CUDA版本
1.需要注意显卡的算力必须与Cuda runtime version相匹配。
在https://en.wikipedia.org/wiki/CUDA#GPUs_supported查看显卡算力和与之相匹配的Cuda runtime version
2.我前面已经在控制面板中查到了

三、本地下载CUDA Toolkit安装包进行安装
官网https://developer.nvidia.com/cuda-toolkit-archive
在这里插入图片描述

选择默认路径
在这里插入图片描述
在这里插入图片描述

选择自定义安装,然后只勾选第一项

在这里插入图片描述
在这里插入图片描述在这里插入图片描述
在这里插入图片描述

(下面这张图不用管,是之前卸载过的安装的版本)
这张图不用管,是之前卸载过的安装的版本
右键“此电脑”,查看环境变量如下,发现已经自动添加好了cuda的路径
在这里插入图片描述

cmd里查看版本信息nvcc -V
在这里插入图片描述

查看CPU运行时的监测界面cd C:\Program Files\NVIDIA Corporation\NVSMI
nvidia-smi
在这里插入图片描述

进入cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.3\extras\demo_suite
输入bandwidthTest.exe
在这里插入图片描述
输入.\deviceQuery.exe
在这里插入图片描述

如果以上两个个命令均看到Result = PASS ,说明安装成功

四、安装cudnn8.9
官网https://developer.nvidia.com/rdp/cudnn-archive
将下载下来的压缩包解压后,将对应文件夹的文件放到CUDA安装路径下的对应文件夹里即可。
在这里插入图片描述
在这里插入图片描述

五、安装PyTorch
官网:https://pytorch.org/
没有符合本机的pytorch版本时点击这个
在这里插入图片描述
浏览直至找到匹配的版本
(由于我的cuda 版本是 12.3,就安装的cuda 12.1,亲测是可以成功匹配的)在这里插入图片描述

打开Anaconda Powershell Prompt,输入对应命令
在这里插入图片描述
然后会有y/n,选择y即可

六、验证PyTorch是否安装成功
打开jupyter notebook,运行一些测试代码来确认您的环境是否按预期工作,例如:

import torch
print(torch.__version__)
print(torch.cuda.is_available())
  • 这将打印出 PyTorch 的版本,并检查 CUDA 是否可用。
  • 输入结果如下图,即为成功
    在这里插入图片描述

七、如何创建、删除环境
不需要删除之前创建的环境。在Anaconda中,您可以同时拥有多个环境,每个环境都有其独立的Python版本和库。这是Anaconda设计的一个强大功能,它允许您为不同的项目创建不同的环境,而不同环境之间不会相互影响。

如果您想创建一个新的环境,使用Python 3.7并在其中安装PyTorch或其他库,您可以直接创建这个新环境,无需删除或更改之前的环境。下面是创建新环境的步骤:

创建新的Python 3.7环境

conda create -n pytorch37 python=3.7

这将创建一个名为pytorch37的新环境,并在其中安装Python 3.7。

激活新环境

conda activate pytorch37

这个命令激活了新创建的环境。

在新环境中安装PyTorch和其他必要的库

根据您的需求,在新环境中安装PyTorch和其他库。由于您提到了特定的PyTorch和CUDA版本,如果您打算在这个新环境中使用不同的版本,您可以选择合适的版本进行安装。例如,如果您想安装与Python 3.7兼容的PyTorch版本,您可能需要找到适合这个Python版本的PyTorch版本。您可以参考PyTorch的官方安装指南来选择合适的命令。

使用新环境

创建并设置好新环境后,您就可以在这个环境中进行开发了。如果您使用Jupyter Notebook,还需要在新环境中安装ipykernel,并将该环境添加为Jupyter的一个内核,如前面所述。

管理多个环境

  • 查看所有可用环境:
conda env list
  • 删除环境(如果需要):
conda remove --name env_name --all

env_name替换为您想要删除的环境名称。

通过使用这些策略,您可以灵活地管理和使用多个环境,以满足不同项目的需求,而无需担心环境之间的冲突。

在您提供的图片中,代码已经列出了可用的Jupyter内核,但它只显示了python3。这意味着在Jupyter中注册的内核中可能没有包含名为pytorch或其他您可能期望的内核。

八、如何查看图片中所示的内核中有没有pytorch呢?(检查特定内核是否存在)

您可以通过以下步骤检查特定内核是否存在:

  1. 终端或Anaconda Prompt:

    • 打开终端(或在Windows上打开Anaconda Prompt)。

    • 激活您想要使用的Conda环境,例如pytorch_env

    • 输入以下命令来列出已注册的内核:

      conda activate pytorch_env
      jupyter kernelspec list
      
  2. Jupyter Notebook:

    • 在Jupyter Notebook中运行以下命令来获取更详细的信息,这将包括内核的名称和它们的安装路径:

      from jupyter_client.kernelspec import KernelSpecManager
      ksm = KernelSpecManager()
      kernel_specs = ksm.get_all_specs()
      
      for kernel_name, kernel_info in kernel_specs.items():
          print(f"Kernel name: {kernel_name}")
          print(f"Path: {kernel_info['resource_dir']}")
      

这将列出所有注册的内核和它们的路径。如果您在列表中看不到名为pytorch或者类似于您为PyTorch环境创建的内核名称,那么您需要确保您已经为该环境安装了ipykernel,并且已经添加了一个内核。

添加缺失的内核

如果您发现pytorch内核确实缺失,您需要执行以下步骤来添加它:

  1. 激活您的环境:

    conda activate pytorch
    
  2. 安装ipykernel(如果尚未安装):

    conda install ipykernel
    
  3. 添加内核:

    python -m ipykernel install --user --name pytorch --display-name "PyTorch (Python 3.6)"
    

完成这些步骤后,重新启动Jupyter Notebook,并检查内核列表,以确认您的pytorch_env内核现在可用。

您已经成功在您的 pytorch 环境中安装了 ipykernel 并且将其作为一个新内核添加到了 Jupyter 中。接下来,您应该做以下几步:

  1. 启动 Jupyter Notebook:

    在 Anaconda Prompt 中,您可以直接启动 Jupyter Notebook。请确保您仍在您的 pytorch 环境中,如果不确定,再次运行 conda activate pytorch

    jupyter notebook
    
  2. 在 Jupyter Notebook 中切换内核:

    • 当您的 Jupyter Notebook 启动并且您打开了一个 notebook 之后,在 notebook 的菜单栏中选择 Kernel > Change kernel,然后从列表中选择刚刚添加的 “PyTorch (Python 3.x)” 内核。
  3. 如果有必要,重新启动 Notebook:

    • 如果在切换内核后遇到问题,尝试在 notebook 的菜单栏中选择 Kernel > Restart
  4. 开始您的项目:

    • 一旦一切设置完毕,并且您已经验证了 PyTorch 和 CUDA 工作正常,您就可以开始在 Jupyter Notebook 中进行您的深度学习项目了。

您现在应该能够在 Jupyter Notebook 中正常使用 PyTorch,并利用您的 GPU(如果 CUDA 可用并且您有支持的 GPU)。

【部分参考:https://blog.csdn.net/weixin_45597212/article/details/135553151】

  • 17
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
安装PyTorch和CUDA的步骤如下: 1. 首先,确保你的计算机上已经安装了NVIDIA显卡驱动程序,并且你的显卡支持CUDA。你可以在NVIDIA官网上下载最新的驱动程序并安装。 2. 接下来,根据你的操作系统选择合适的PyTorch版本。你可以在PyTorch官网上找到各种版本的安装指南。选择一个与你的操作系统和CUDA版本兼容的版本。 3. 如果你还没有安装Anaconda,建议先安装Anaconda。你可以在Anaconda官网上下载相应的安装程序并按照指示进行安装。 4. 打开Anaconda Prompt(Windows)或终端(Linux/MacOS)。 5. 创建一个新的虚拟环境(可选但推荐),这可以帮助你在不同的项目之间隔离依赖关系。在命令行中运行以下命令: ``` conda create -n pytorch_env python=3.8 ``` 6. 激活虚拟环境。在命令行中运行以下命令: ``` conda activate pytorch_env ``` 7. 在激活的环境中,使用conda或pip安装PyTorch和CUDA。根据你选择的版本和操作系统,运行以下命令之一: - 使用conda安装: ``` conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge ``` - 使用pip安装: ``` pip install torch torchvision torchaudio ``` 8. 安装完成后,你可以尝试导入PyTorch来验证是否安装成功。在Python交互式环境中运行以下代码: ```python import torch print(torch.__version__) ``` 如果一切顺利,你应该能够看到已安装PyTorch版本号。现在你已经成功安装PyTorch和CUDA,可以开始使用它们进行深度学习任务了。请注意,安装过程可能因操作系统、CUDA版本和PyTorch版本而有所不同,请根据实际情况进行调整。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值