在conda虚拟环境安装pytorch

参考官网
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

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