Pytorch1.5.1 cuda10.2版本python3.7安装方法

甜度超标°

已于 2022-10-26 08:40:57 修改

阅读量1.9k

点赞数 1

分类专栏：学习笔记文章标签：深度学习 pytorch python

于 2022-08-21 18:31:32 首次发布

本文链接：https://blog.csdn.net/qq_38703529/article/details/126453612

版权

学习笔记专栏收录该内容

83 篇文章 6 订阅

订阅专栏

Pytorch1.5.1 cuda10.2版本python3.7安装方法

安装步骤

安装步骤

1.下载pytorch。在清华源https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/?C=M&O=D下载对应pytorch版本，pytorch-1.5.1-py3.7_cuda102_cudnn7_0.tar.bz2。
在这里插入图片描述
2.下载torchvision，torchvision===0.6.1。

3.进入下载路径进行离线安装。打开conda prompt,输入

cd E:\data\torch_bag #改成自己的下载路径

离线安装，

conda install --offline pytorch-1.5.1-py3.7_cuda102_cudnn7_0.tar.bz2

在这里插入图片描述

conda install --offline torchvision-0.6.1-py37_cu102.tar.bz2

在这里插入图片描述
4.安装cudatoolkit=10.2

conda install cudatoolkit=10.2

在这里插入图片描述
5.检验是否安装成功

import torch
from torch.backends import cudnn

x = torch.Tensor([1.0])
xx = x.cuda()
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
print(xx)
print(cudnn.is_acceptable(xx))

在这里插入图片描述