小白虚拟环境安装cuda版本的pytorch并检查是否安装成功

如果电脑自带显卡的话

安装pytorch一定要和自身电脑所带的cuda版本兼容

尽量安装在虚拟环境中

1、激活虚拟环境

如果有不会设置虚拟环境的宝子可以参照我之前小白使用conda在windows11下D盘安装虚拟环境的记录-优快云博客

conda activate pytorch

2、查看cuda版本

nvidia-smi

在右上角可以看到 cuda版本是12.0 

进入到pytorch官网中找到相应的安装版本

根据相应电脑配置获取安装指令

提醒一下：尽量安装与cuda版本相应的或者较低版本的

我的cuda是12.0版本，选择cuda11.8的版本进行安装

如果有pip3  直接复制代码

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

如果没有pip3 就替换成pip即可

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

 这样pytorch就成功安装好了

3、检查是否安装成功

3.1 查看安装的包

conda list

可以看到安装的是cuda11.8版本下的2.1.2版本的torch

3.2 使用代码查验

查看torch版本

import torch
print(torch.__version__)

或者 查看是否安装好 运行结果为True 就是安装好了

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