如何使用Anaconda安装Pytorch1.13.0 GPU环境(CUDA11.6版)

作者:Nicky2024.01.17 11:06浏览量:41

简介:本文将指导你如何使用Anaconda安装PyTorch 1.13.0 GPU环境,并支持CUDA 11.6。我们将分步骤详细介绍如何配置Python环境、安装依赖项、设置CUDA和cuDNN版本,以及验证安装是否成功。

千帆应用开发平台“智能体Pro”全新上线 限时免费体验

面向慢思考场景,支持低代码配置的方式创建“智能体Pro”应用

立即体验

首先,确保你已经安装了Anaconda。你可以从Anaconda官网下载并安装适合你操作系统的版本。一旦Anaconda安装完成,你可以创建一个新的虚拟环境。打开终端或命令提示符,输入以下命令:
conda create -n myenv python=3.8
这将创建一个名为“myenv”的新虚拟环境,并使用Python 3.8版本。接下来,激活新创建的虚拟环境。在Windows上,输入以下命令:
activate myenv
在macOS和Linux上,输入以下命令:
conda activate myenv
现在,你已经激活了虚拟环境。接下来,我们需要安装PyTorch和CUDA。在Anaconda终端中,输入以下命令:
conda install pytorch=1.13.0 torchvision torchaudio -c pytorch
这将安装PyTorch 1.13.0版本以及torchvision和torchaudio包。注意,我们使用了“-c pytorch”参数来指定从PyTorch的频道安装。
接下来,安装CUDA 11.6。由于Anaconda本身不提供CUDA 11.6的安装包,你需要从NVIDIA官网下载并安装CUDA 11.6。安装完成后,设置环境变量以确保系统能够找到CUDA。具体步骤取决于你的操作系统,但一般来说,你需要将CUDA的bin、lib和lib64目录添加到PATH环境变量中。在Windows上,你可以在系统属性中的高级选项卡下编辑环境变量;在macOS和Linux上,你可以编辑~/.bashrc或~/.bashprofile文件并添加以下行:
export PATH=/path/to/cuda/bin:$PATH
export LDLIBRARY_PATH=/path/to/cuda/lib:$LD_LIBRARY_PATH
将“/path/to/cuda”替换为你的CUDA安装路径。保存文件后,重新启动终端或命令提示符以使更改生效。
现在,你已经成功安装了PyTorch和CUDA。最后一步是验证安装是否成功。打开Python解释器或Jupyter Notebook,并尝试导入PyTorch库:
import torch
print(torch.__version
)
如果一切顺利,你将看到PyTorch的版本号(在这种情况下是1.13.0)被打印出来。另外,你可以通过以下代码检查CUDA是否正常工作:
import torch.cuda as cuda
print(cuda.is_available())
如果看到True被打印出来,那么说明你已经成功安装了PyTorch 1.13.0 GPU环境并支持CUDA 11.6。

article bottom image

相关文章推荐

发表评论