0
点赞
收藏
分享

微信扫一扫

Cuda+Cdnn+Torch安装Torch.Cuda.is_available =True注意要点(踩坑记录)

半夜放水 2022-04-13 阅读 49
(env1_yolov5) E:\PY\YOLOV5\yolov5-6.0>pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Looking in links: https://download.pytorch.org/whl/torch_stable.html
Collecting torch==1.10.1+cu102
  Downloading https://download.pytorch.org/whl/cu102/torch-1.10.1%2Bcu102-cp39-cp39-win_amd64.whl (1485.9 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.5/1.5 GB 3.4 MB/s eta 0:00:00
Collecting torchvision==0.11.2+cu102
  Downloading https://download.pytorch.org/whl/cu102/torchvision-0.11.2%2Bcu102-cp39-cp39-win_amd64.whl (2.6 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.6/2.6 MB 4.1 MB/s eta 0:00:00
Collecting torchaudio==0.10.1
  Downloading https://download.pytorch.org/whl/cu113/torchaudio-0.10.1%2Bcu113-cp39-cp39-win_amd64.whl (336 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 336.6/336.6 KB 6.9 MB/s eta 0:00:00
Requirement already satisfied: typing-extensions in e:\anaconda3\envs\env1_yolov5\lib\site-packages (from torch==1.10.1+cu102) (4.1.1)
Requirement already satisfied: pillow!=8.3.0,>=5.3.0 in e:\anaconda3\envs\env1_yolov5\lib\site-packages (from torchvision==0.11.2+cu102) (9.1.0)
Requirement already satisfied: numpy in e:\anaconda3\envs\env1_yolov5\lib\site-packages (from torchvision==0.11.2+cu102) (1.22.3)
Installing collected packages: torch, torchvision, torchaudio
  Attempting uninstall: torch
    Found existing installation: torch 1.11.0
    Uninstalling torch-1.11.0:
      Successfully uninstalled torch-1.11.0
  Attempting uninstall: torchvision
    Found existing installation: torchvision 0.12.0
    Uninstalling torchvision-0.12.0:
      Successfully uninstalled torchvision-0.12.0
  Attempting uninstall: torchaudio
    Found existing installation: torchaudio 0.11.0+cu113
    Uninstalling torchaudio-0.11.0+cu113:
      Successfully uninstalled torchaudio-0.11.0+cu113
Successfully installed torch-1.10.1+cu102 torchaudio-0.10.1+cu113 torchvision-0.11.2+cu102

(env1_yolov5) E:\PY\YOLOV5\yolov5-6.0>python
Python 3.9.7 (default, Sep 16 2021, 16:59:28) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.10.1+cu102'
>>> torch.cuda.is_available()
True

总结:

0.参考(1条消息) Cuda和cuDNN安装教程(超级详细)_jhsignal的博客-CSDN博客_cuda安装教程

1.安装Cuda注意点

       1.1  Version是windows系统,选择win10

       1.2 exe(local)是完全离线安装包,可以复制链接用迅雷下载,速度很快.如果选择(network)的话会安装到一半再网上下载,可能速度很慢.

       1.3  点击安装包后的第一个弹窗,建议在其他盘新建一个空文件夹,安装结束后会自动删除掉

         1.4 记住安装位置,后面需要将文件放入,最好使用默认位置.

         1.5 出现图形程序问题不用担心,因为图形程序版本高能兼容低版本Cuda.

(1条消息) nvidia驱动,cuda与cudnn的关系_「已注销」的博客-CSDN博客_cuda和nvidia驱动的关系

        1.6 按照教程设置系统变量(按照教程一步步来,该重启就重启,保险一点!)

2.安装cuDNN注意点

        2.1 按照Cuda版本选择cuDNN, 不要被显卡程序里面的Cuda.dll和nvidia-sim显示的版本误导了,版本号从nvcc -V/安装包截图/安装路径 确定!

         2.2 移动文件夹,其中lib文件夹中有两个子文件夹,不用担心,直接放在lib下就可以,重要的是后面添加path环境变量(我不知道父文件夹和两个子文件夹都复制了一份)

        2.3 按照教程将上面三个文件夹添加进path环境变量

3.按照教程检查安装是否安装正确.

4.安装torch

        4.1 进入pycharm,进入环境

        4.2 进入Previous PyTorch Versions | PyTorch,直接找到轮子-对应cuda版本,复制命令,

在pycharm中粘贴运行.

         4.3 运行命令,检查是否可用

>>> import torch
>>> torch.__version__
'1.10.1+cu102'
>>> torch.cuda.is_available()
True

 

 

 

举报

相关推荐

0 条评论