使用Anaconda 管理环境,直接搜索下载pytorch包,能够看到版本信息,却在pycharm中无法使用。
查找发现系统没装Tootkit,应该是上次重装系统后没搞了。

1. 查看当前设备是否具有CUDA Toolkit

nvidia-smi


这里显示CUDA版本 11.7

查看对应CUDA Toolkit版本

nvcc -V

2. 当前设备没有CUDA Toolkit

查找对应CUDA Toolkit

win搜索栏搜索NVIDIA
打开NVIDIA Control Panel
在左下角系统信息中,可以看到驱动程序版本,本机GTX1070 对应版本:516.94
同样,在组件栏中可以找到
CUDA版本

官网查表找到对应CUDA Toolkit版本
Toolkit对应表

下载CUDA Toolkit

确定好所需CUDA Toolkit版本,在官网下载,这里选择11.7.1
下载链接
下载后按流程安装

验证是否安装成功

cmd 输入 nvcc -V

成功

3. 找到Pytorch版本并安装

Pytorch官网

在conda中打开环境命令行界面
!!注意使用管理员权限!!

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

4. 检查是否安装成功

import torch
x = torch.rand(5, 3)
print(x)

检查是否可以使用当前GPU

import torch
torch.cuda.is_available()