使用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 Toolkit版本
下载CUDA Toolkit
确定好所需CUDA Toolkit版本,在官网下载,这里选择11.7.1
下载后按流程安装
验证是否安装成功
cmd 输入 nvcc -V
成功
3. 找到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()