由于之前装了个假的gpu版本,所以来记录一下正确的怎么安装。
先看NVIDIA控制面板-帮助-系统信息-组件
看驱动版本,我的是9.0.
或者是 cmd输入NVIDIA-smi看 我的版本是385.54,然后看cuda toolkit对应版本,能装cuda9.0
一、CUDA Tollkit下载:https://developer.nvidia.com/cuda-toolkit-archive
然后运行安装
二、对应版本的CUDNN下载:https://developer.nvidia.com/rdp/cudnn-download
然后重命名 为cudnn复制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0下面
然后配置环境变量,按照PyTorch(GPU版)详细安装教程_哔哩哔哩_bilibili
cuda装完之后cmd,nvcc -V验证一下
三、pytorch安装
创建虚拟环境 conda create -n pytorch python=3.7
activate pytorch
然后是重点,pytorch下载从官网对应版本找命令容易出错。Previous PyTorch Versions | PyTorch找到页面最后能找到cuda9.0 对应的,下载后pip
进去找到对应的torch和torchvision
然后进入到这两个文件所在的文件夹,E: ,cd E:bishepytorch
pip install whl文件
最后进入python ,torch.cuda.is_available()是true,就成功了。



