Pytorch和cuda环境安装

为了避免更换不同版本Cuda的麻烦，环境在Anaconda里面搭建。

创建conda环境

Anaconda的安装就不赘述，首先是创建一个虚拟环境：

1	conda create --name your-env-name python=3.13

your-env-name是环境名称，随意更换，python版本也可根据不同环境进行更改。这里要注意--name是两个-，如果只打一个-你有概率会获得一个名字叫ame的虚拟环境😂

千万不要忘了进入虚拟环境再进行下面的操作：

1	conda activate your-env-name

安装Cuda

这里有两种方法，第一是用conda命令进行安装，源应该是conda仓库。第二种是通过pip，源是pip的仓库。其中conda安装的命令虽然更简单，但是conda仓库里Cuda版本十分有限，所以如果conda仓库里如果没有自己想要的版本还是推荐用pip进行安装。

通过conda安装

可以通过以下命令查看conda仓库中有哪些版本的Cuda：

1	conda search cudatoolkit --info

通过conda安装Cuda的命令为：

1	conda install cudatoolkit=11.7

通过pip安装

NVIDIA 官网提供了通过pip安装的方法。

安装命令如下：

# 更新pip
py -m pip install --upgrade setuptools pip wheel
# 安装nvidia-pyindex
py -m pip install nvidia-pyindex
# 安装cuda12
py -m pip install nvidia-cuda-runtime-cu12
# 可选 安装其他模块
py -m pip install nvidia-<library>

其中py需要替换成python。

安装Pytorch

Pytorch 官网提供了对应的安装命令

这里贴出我使用的：

1	pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

检验安装

直接在命令行中输入python进入到python，输入以下几行

1
2
3

import torch
print(torch.version.cuda)
torch.cuda.is_available()

注意import之后会有一段时间等待python加载。

PS：今天也是我第一次用anaconda进行搭建这个环境，仅仅是搭建还未投入到代码的使用，可能会有问题。

PS2：封面是大空スバル，两天前是她的生日！しゅばしゅばしゅば！！！