初次安装Pytorch过程

第一次安装Pytorch,刚开始安装的时候装错了CUDA的版本号

这里最高支持12.2.138, 但是我装了一个12.2.140的CUDA,导致不兼容我在测试时发现

cpp 复制代码
import torch

# if torch.cuda.is_available():
#     print("GPU可用")
# else:
#     print("GPU不可用")


# current_device = torch.cuda.current_device()
# device_name = torch.cuda.get_device_name(current_device)
# print(f"当前GPU设备索引: {current_device}")
# print(f"当前GPU设备名称: {device_name}")

print("PyTorch版本:", torch.__version__)
print(torch.cuda.is_available())

# print("PyTorch版本:", torch.__version__)

# # 创建一个简单的张量
# x = torch.tensor([1.0, 2.0, 3.0])
# print("张量x:", x)

输出是False(GPU不可用),后来检查中发现是CUDA版本不兼容,所以我又把之前的pytorch全部卸载了,又重新装的低版本的CUDA,然后再装pytorch,最后成功了,显示True(GPU可用),写这篇博客的意义在于记录一下安装Pytorch的过程,避免以后遗忘。

过程:

1.首先在命令行窗口中,输入

csharp 复制代码
python --version

查看自己的python版本

2.

安装PyTorch通常涉及选择正确的PyTorch版本以及安装适合您的操作系统和硬件配置的PyTorch版本。以下是在常见操作系统上安装PyTorch的一般步骤:

注意:PyTorch的安装方式和要求可能会随时间和版本的变化而有所不同。确保在安装之前查看PyTorch官方网站以获取最新的安装说明和要求。

使用pip安装PyTorch(CPU版本)

如果您只想在CPU上使用PyTorch,可以使用pip来安装它。以下是一般步骤:

1.打开命令行或终端窗口。

2.在终端中输入以下命令:

cpp 复制代码
pip install torch

这将安装最新版本的PyTorch(CPU版本)。

使用conda安装PyTorch(CPU版本)

如果您使用Anaconda或Miniconda作为Python环境管理器,可以使用conda来安装PyTorch。以下是一般步骤:

1.打开命令行或终端窗口。

2.创建一个新的conda环境(可选,但推荐):

cpp 复制代码
conda create -n myenv python=3.8  # 创建一个名为myenv的Python 3.8环境
conda activate myenv  # 激活新环境

3.安装PyTorch(CPU版本):

cpp 复制代码
conda install pytorch torchvision torchaudio cpuonly -c pytorch

使用pip安装PyTorch(GPU版本)

如果您要在支持CUDA的GPU上使用PyTorch,可以使用pip来安装PyTorch GPU版本。首先,确保您的系统上已经安装了NVIDIA驱动和CUDA工具包。

1.打开命令行或终端窗口。

2.在终端中输入以下命令:

cpp 复制代码
pip install torch torchvision torchaudio

这将安装最新版本的PyTorch(GPU版本),如果您的GPU和CUDA版本与PyTorch兼容的话。

使用conda安装PyTorch(GPU版本)

如果您使用Anaconda或Miniconda,并且要在支持CUDA的GPU上使用PyTorch,可以使用conda来安装PyTorch。同样,请确保您的系统上已经安装了NVIDIA驱动和CUDA工具包。

1.打开命令行或终端窗口。

2.创建一个新的conda环境(可选,但推荐):

cpp 复制代码
conda create -n myenv python=3.8  # 创建一个名为myenv的Python 3.8环境
conda activate myenv  # 激活新环境

3.安装PyTorch(GPU版本):

cpp 复制代码
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

请注意,上述命令中的cudatoolkit版本应与您的CUDA版本匹配。您可以使用nvcc --version命令来查看您的CUDA版本。

安装完成后,您可以在Python环境中导入PyTorch并开始使用它。记得查看官方文档以获取更多信息和示例代码。


以上是安装Pytorch的过程,但是在此之前我们需要先安装CUDA,具体操作可以参考👉安装CUDA这篇文章。
CUDA下载网站

进去之后点这里

下载旧版本的CUDA(一般不要下载最新版,因为会有版本不兼容的问题)

这里我下的是11.8.0

后续的步骤就可以参考👉安装CUDA这篇文章。

ps:当前最新的cudatoolkit为11.8.0,查询网站👉cudatoolkit

下完CUDA后就需要下载Pytorch了,下载网站👉Pytorch下载网站

切记:下载pytorch的版本要参考着CUDA来下载,否则会导致二者不兼容。

下面是我下载pytorch的配置:

因为我的CUDA就是11.8.0的,所以我选择了CUDA 11.8这个选项。

之后复制他的command,去vscode的虚拟环境里下载(命令行窗口应该可以,但是一定也要开启虚拟环境),具体开启虚拟环境的步骤

cpp 复制代码
conda activate myenv	(激活虚拟环境)
conda init powershell	(初始化powershell环境)

一般只需要运行第一行的代码即可开启虚拟环境

开启虚拟环境后前方会出现(myenv)的字样

例如:

这就代表着已经成功开启虚拟环境了,然后再在虚拟环境里用刚才复制的command代码来下载Pytorch,例如:

然后耐心等待下载完成即可。

要检查是否成功安装了PyTorch,您可以打开Python解释器(在命令行或终端中运行python命令)并尝试导入PyTorch。如果没有出现导入错误,那么您已成功安装PyTorch。

以下是一些示例代码,演示如何检查PyTorch的安装:

python 复制代码
import torch

# 检查PyTorch版本
print("PyTorch版本:", torch.__version__)

# 创建一个简单的张量
x = torch.tensor([1.0, 2.0, 3.0])
print("张量x:", x)

如果您在运行这些代码时没有看到任何错误,并且能够成功导入PyTorch并创建张量,那么您的PyTorch安装就是成功的。

另外,您还可以运行以下命令来检查PyTorch是否已安装以及其版本:

cpp 复制代码
python -c "import torch; print(torch.__version__)"

这将在命令行中输出PyTorch的版本号。如果成功显示版本号而没有导入错误,那么PyTorch已经成功安装并可用。

如果您在安装或导入PyTorch时遇到任何错误,可能需要检查安装过程中是否出现问题或根据错误消息解决问题。确保遵循官方文档和安装说明以确保正确的安装。

还可以通过运行代码

python 复制代码
import torch

# if torch.cuda.is_available():
#     print("GPU可用")
# else:
#     print("GPU不可用")


current_device = torch.cuda.current_device()
device_name = torch.cuda.get_device_name(current_device)
print(f"当前GPU设备索引: {current_device}")
print(f"当前GPU设备名称: {device_name}")

print("PyTorch版本:", torch.__version__)
print(torch.cuda.is_available())

# print("PyTorch版本:", torch.__version__)

# # 创建一个简单的张量
# x = torch.tensor([1.0, 2.0, 3.0])
# print("张量x:", x)

来判断CUDA是否可以正常使用,如果输出

则代表你已经成功安装Pytorch和与之相匹配的CUDA了!

最后贴几个对我帮助很大的博客以及网站:
超详细GPU部署 (pytorch+tensorflow)博客
安装CUDA博客
pytorch 下载安装全流程详细教程
超链接下载地址(没用到)
CUDA 11.8.0下载地址
Pytorch下载地址

检查当前已安装的CUDA版本

查看电脑可以安装的CUDA最高版本,即你所安装的CUDA只能≤12.2.138(每人情况不同)

相关推荐
SsummerC1 小时前
【leetcode100】数组中的第K个最大元素
python·算法·leetcode
伊玛目的门徒1 小时前
解决backtrader框架下日志ValueError: I/O operation on closed file.报错(jupyternotebook)
python·backtrader·量化·日志管理·回测
java1234_小锋2 小时前
一周学会Pandas2 Python数据处理与分析-编写Pandas2 HelloWord项目
python·pandas·python数据分析·pandas2
凯强同学3 小时前
第十四届蓝桥杯大赛软件赛省赛Python 大学 C 组:7.翻转
python·算法·蓝桥杯
小西几哦5 小时前
3D点云配准RPM-Net模型解读(附论文+源码)
人工智能·pytorch·3d
独好紫罗兰5 小时前
洛谷题单3-P1217 [USACO1.5] 回文质数 Prime Palindromes-python-流程图重构
开发语言·python·算法
1alisa5 小时前
Pycharm v2024.3.4 Windows Python开发工具
ide·python·pycharm
独好紫罗兰5 小时前
洛谷题单2-P1424 小鱼的航程(改进版)-python-流程图重构
开发语言·python·算法
程序员小赵同学6 小时前
AI Agent设计模式二:Parallelization
开发语言·python·设计模式
杰克逊的日记6 小时前
CentOs系统部署DNS服务
linux·python·centos·dns