1.先查看一下版本:
2.控制面板里卸载原来的10.0
留下:NVIDIA的图形驱动程序、NVIDIA Physx系统软件、NVIDIA GeForce Experience
3.注册表卸载,用腾讯电脑管家或360即可
4. 下载cuda 10.2,注意选择exe,用迅雷加速
5. 默认即可
6.
7.
8.
9.建议默认安装在C盘 。
10.
11.
有这个nvcc.exe就说明CUDA安装已成功
12.
CUDNN的下载
官网下载地址:https://developer.nvidia.com/rdp/cudnn-download
先注册一个账户并填写完信息
13.
下载之后,解压缩,将CUDNN文件夹里面的bin、include、lib文件复制粘贴到cuda安装目录的对应文件夹下
不知道cuda安装在哪个目录可以用 set cuda查看。
14.
使用pip卸载Pytorch
pip uninstall torch
pip uninstall torchvision
下来参考这篇博客安装pytorch1.6和torchvision0.7即可
验证:输入python进入python环境。
导入torch:import torch
查看版本:
import torch
print(torch.__version__)
print(torch.cuda.is_available())