赞
踩
本文用来预防pytorch出问题需要重装:(
一、下载安装对应版本CUDA和cudnn
(之前下载的,现在忘记记录在哪里了,找到记录再放,或者去参看其他博客)
版本选择参考:
GPU版本的PyTorch安装与环境配置_ke996的博客-CSDN博客_安装gpu版本的pytorch
安装结果
二、下载torch,并安装
1.下载地址https://download.pytorch.org/whl/torch_stable.html
2.版本对应关系
(1)CUDA & PyTorch
(2)选择python对应版本,或者安装对应版本python(后来因参考代码需要改下1.8.0版本)
3.安装
cmd到下载目录下,使用pip install
三、测试使用
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。