参考官网
https://pytorch.org/get-started/previous-versions/

安装后检查是否安装成功

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

用conda安装pytorch可能出现cuda不能用的情况,建议用pip

~~pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116~~ 

在这里插入图片描述
下载的时候看清楚是否是gpu版本,第一次下载的pytorch是cpu版本的,卸载后安装的一直都是cpu,即使后面跟的是cuda,也不会安装成gpu的!!
在这里插入图片描述


首先去清华开源库下载pytorch的gpu版,存放在如下地址中:E:\Software\CUDA10.2\pytorch-1.7.0-py3.8_cuda102_cudnn7_0.tar
 
(1)安装pytorch gpu版
 
conda install E:\Software\CUDA10.2\pytorch-1.7.0-py3.8_cuda102_cudnn7_0.tar.bz2
 
 
(2)安装其他选项
 
conda install torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 cudnn=7.6.5

非常感谢这位老哥孜孜不倦地刨根问底!

更多推荐