PyTorch 学习by_LX（一）：pytorch GPU版本安装等碎碎念

2024-04-22 13:52:17

新买的RTX2070到了，一顿操作后装的是cuda11.0，而pytorch目前支持的最高cuda版本为10.2；无奈需要重装，本想在不卸载cuda11.0的条件下安装10.2，一顿操作猛如虎，结果运行pytorch还是报错，无奈选择卸载11.0；

1.卸载cuda11.0

进入控制面板，卸载程序，将带有11.0版本号的NVIDIA CUDA逐个卸载(本人卸载时忘了截图了，这里截图以10.2为例)

打开计算机-属性-高级系统设置-环境变量-系统变量，将系统变量中存在CUDA11.0的路径都删除

2.安装cuda10.2

CUDA下载：https://developer.nvidia.com/cuda-toolkit-archive

下载工具包，安装CUDA10.2
参考资料：PyTorch（GPU版）详细安装教程.

安装完成后，输入nvcc -V检查是否安装成功；

3.安装CUDNN

关于为什么要安装CUDNN，参考以下资料
显卡，显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn到底是什么？.

CUDNN下载：https://developer.nvidia.com/rdp/cudnn-download，这里需要注册NVIDIA才可以下载，

下载完成后，解压文件，改名为CUDNN文件，剪贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2中

4.安装pytorch

下载命令行在官网可以看到：https://pytorch.org/

找到对应版本，命令行输入下载

下载完成后，torch.cuda.is_available()显示True即安装成功。

踩过好多坑，终于下载成功了~~

PyTorch 学习by_LX（一）：pytorch GPU版本安装等碎碎念相关推荐

pytorch，torch，torchvision的gpu版本安装避坑
pytorch,torch,torchvision的gpu版本安装避坑在线or离线安装 pytorch torch和torchvision 在线or离线安装最近使用pytorch跑深度学习模型,需 ...
Pytorch：入门指南和 PyTorch 的 GPU版本安装(非常详细)
Pytorch: 入门指南和 PyTorch 的 GPU版本安装(非常详细) Copyright: Jingmin Wei, Pattern Recognition and Intelligent S ...
Pytorch学习：Task4 PyTorch激活函数原理和使用
Pytorch学习:Task4 PyTorch激活函数原理和使用 1.torch.nn.ELU Sigmoid和ReLU结合体,具有左侧软饱和性 2.torch.nn.LeakyReLU 数学表达式: ...
PyTorch学习笔记：PyTorch初体验
PyTorch学习笔记:PyTorch初体验一.在Anaconda里安装PyTorch 1.进入虚拟环境mlcc 2.安装PyTorch 二.在PyTorch创建张量 1.启动mlcc环境下的Spy ...
TensorFlow GPU 版本安装个人总结：Win10 + Python3.5 + CUDA 9.0.176 + cudnn v7.5.0.56 + TensorFlow 1.12.0
TensorFlow GPU 版本安装个人总结:Win10 + Python3.5 + CUDA 9.0.176 + cudnn v7.5.0.56 + TensorFlow 1.12.0 接触机器学 ...
pytorch gpu版本安装
1,安装anaconda 2,查看显卡版本信息 3,安装对应的驱动程序cuda tooltik 4,安装pytorch 首先安装anaconda ,安装anaconda的目的在于创建虚拟环境,因为有的 ...
pytorch的cuda环境搭建（GPU版本安装）
第一步:安装cuda 安装cuda,这里还需要安装对应版本的Visual Studio(具体参考本博主的博客 https://blog.csdn.net/qq_36653505/article/det ...
pytorch GPU版本安装 win10
我原先装的是pytorch cpu版本,在复现论文源码的时候提示AssertionError: Torch not compiled with CUDA enabled,查了一下资料,又看了一下电脑的 ...
GPU版本安装Pytorch教程最新方法
目录步骤第一步:安装 Anaconda 和 Pycharm 软件第二步:下载安装CUDA11.3 (1)首先查看自己电脑GPU版本方式一:搜索框输入nvidia,打开nvidia控制面板方式 ...

最新文章

热门文章