0
点赞
收藏
分享

微信扫一扫

【12321骚扰电话举报受理中心-短信验证安全分析报告】

RJ_Hwang 2024-07-24 阅读 29

1,最终结果

import torch

print(torch.cuda.is_available()) #显示True,则安装成功
print(torch.__version__)#打印当前PyTorch版本号。
print(torch.version.cuda)#打印当前CUDA版本号。
print(torch.backends.cudnn.version())# 打印当前cuDNN版本号。
print(torch.cuda.get_device_name(0))# 打印第一个GPU设备的名称。
True
2.2.2
11.8
8700
NVIDIA GeForce RTX 2080 Ti

2,要实现这个结果非常复杂,很多环境不兼容等问题,今天这里一一解答
检查本地硬件驱动

nvidia-smi

在这里插入图片描述
运行 nvcc -V 可知,当前cuda版本为 11.8
在这里插入图片描述
3,下载 cuda11.8:CUDA Toolkit Archive | NVIDIA Developer
在这里插入图片描述
4,下载后根据截图自定义安装,

在这里插入图片描述
5,配置环境变量
在这里插入图片描述
6,下载cudnn:cuDNN Archive | NVIDIA Developer

下载前要先注册信息,不算特别麻烦,勾几个√就行。

在for CUDA 11.x 里面随便选一个版本,我选的是最新版本

在这里插入图片描述
下载完成后,将压缩包解压,将 bin、include、lib 3个文件夹复制到cuda的安装目录v11.8下。

7,安装pytorch2.0
进入pytorch官网:PyTorch

在这里插入图片描述
选择对应的版本安装
这里先要建立虚拟环境

conda create -n gubanjie python==3.9
//激活环境
conda activate gubanjie
//卸载
//conda uninstall torch torchvision
//pip uninstall torch torchvision
//清除缓存
pip cache purge
//然后安装
conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=11.8 -c pytorch -c nvidia
//或者
conda install pytorch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 pytorch-cuda=11.8 -c pytorch -c nvidia
//或者
pip install 
//安装torch
pip install torch torchvision torchaudio

在这里插入图片描述
在这里插入图片描述
搞定在这里插入图片描述

举报

相关推荐

0 条评论