pytorch的安装详细步骤

本文介绍了如何详细安装PyTorch,包括Anaconda、CUDA、cuDNN和PyTorch本身。从Anaconda的安装到CUDA和cuDNN的下载与验证,再到PyTorch的pip安装和环境配置,每个步骤都有清晰的指导,确保GPU加速功能的正常使用。
摘要由CSDN通过智能技术生成

PyTorch是一个非常有可能改变深度学习领域前景的Python库,由FaceBook支持更新。由于其相较于TensorFlow等机器学习框架的易用性与易读性而大火。在当前的机器学习、深度学习等领域内的论文发表重,使用PyTorch进行实验的论文迅速增加。我尝试使用了几星期PyTorch,然后被它的易用性所震惊,在我使用过的各种深度学习库中,PyTorch是最灵活、最容易掌握的。

需要安装的库

  • Anaconda(包管理工具,推荐安装)
  • cuda
  • cuDNN
  • PyTorch

Anaconda

Anaconda的安装这里不再赘述,正常安装即可

cuda

下载安装

cuda官网网址

安装cuda之前,我们需要首先前往PyTorch官方网址查看PyTorch支持的cuda版本
在这里插入图片描述
可以看到当前的PyTorch框架支持9.2和10.1的cuda版本,这里我选择的是9.2版本

这里需要说明的是cuda与cuDNN均为GPU计算框架,即使稍显落后的移动版GPU争对科学计算也可以达到超过CPU5倍的运算效率,推荐安装,你可以点击此处检查你的显卡是否支持cuda以及对应的cuda版本1

在下载cuda需要注意的时,可能你想要下载的cuda版本不是最新的cuda版本,而你在百度搜索cuda时默认第一个搜索结果均会是当前最新的cuda版本,而且很难在NVIDIA Developer官网上找到丰富的版本系列,所以推荐直接在百度搜索cuda时加上对应的版本号
在这里插入图片描述
在这里插入图片描述
进入对应版本的cuda下载网址,选择好你相应的环境下载安装即可(推

以下是PyTorch安装的步骤[^1]: 1. 找到相应版本:根据你的需求,找到适合的PyTorch版本。你可以在PyTorch的GitHub页面(https://github.com/pytorch/vision#installation)上查找版本连接,或者在PyTorch官方网站(https://pytorch.org/)上查看环境命令。 2. 下载步骤cpu版本: 2.1 选择下载版本:根据你的需求选择合适的版本,例如torch 1.12.0、torchvision 0.13.0和Python 3.7。 2.2 创建虚拟环境:使用conda命令创建一个新的虚拟环境,例如`conda create -n env_torch python==3.7`。 2.3 进入虚拟环境:使用conda命令进入新创建的虚拟环境,例如`conda activate env_torch`。 2.4 安装PyTorch:使用conda命令安装PyTorch和相关依赖,例如`conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio cpuonly -c pytorch`。 2.5 验证是否安装成功:在Python环境中导入torch模块,并打印PyTorch的版本号,例如: ```python import torch print(torch.__version__) # 输出安装PyTorch版本号 ``` 3. 安装ipykernel(可选):如果你想在Jupyter Notebook中使用PyTorch,可以安装ipykernel,以便将虚拟环境添加到Jupyter的内核列表中。 4. 将新建的环境写入notebook的kernel中:使用命令`python -m ipykernel install --user --name=env_torch`将新建的虚拟环境添加到Jupyter的内核列表中。 5. 在虚拟环境中打开Jupyter:使用命令`jupyter notebook`在虚拟环境中打开Jupyter Notebook。 6. 选择创建的虚拟环境创建文件:在Jupyter Notebook中选择新建的虚拟环境作为内核,并创建文件。 7. 在Jupyter中测试PyTorch完成:在Jupyter Notebook中编写PyTorch代码,并运行以测试PyTorch是否成功安装
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值