开始需要安装anaconda
参考：
https://blog.csdn.net/wq_ocean_/article/details/103889237?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522165880942016782395397561%2522%252C%2522scm%2522%253A%252220140713.130102334…%2522%257D&request_id=165880942016782395397561&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2alltop_positive~default-1-103889237-null-null.142v34pc_rank_34,185v2control&utm_term=Anaconda&spm=1018.2226.3001.4187
接着
安装CUDA前需要知道自己电脑是否支持及其Nvidia 的版本信息。
调出cmd
输入：nvidia-smi
在图中可以看到当前电脑适配的CUDA版本。
接着
在pytorch官网可以下载对应CUDA的pytorch。
https://pytorch.org/get-started/locally/
在anaconda prompt 里激活PyTorch虚拟环境
conda activate PyTorch
在PyTorch环境中粘贴上图Run this Command
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
对于torch.cuda.is_available() 结果
False
参考以下大神文章：
https://blog.csdn.net/qq_46126258/article/details/112708781
在pycharm中使用PyTorch环境
参考链接：
https://blog.csdn.net/qq_46092061/article/details/119155636
最后在pycharm中测试
import torch
#import torch vision
import numpy as np
arr = np.ones((3,3))
print(“arr的数据类型为：”+str(arr.dtype))
t = torch.tensor(arr)
print(t)
print(torch.version)
print(torch.version.cuda)
print(torch.cuda.is_available())
结果
总算True了
