三、算力、CUDA Driver Version, CUDA Runtime Version、以及安装pytorch

1.首先,确定自己显卡的算力---确定自己的显卡型号

       点击任务管理器---性能

2.CUDA官网----https://en.wikipedia.org/wiki/CUDA----确定显卡算力(8.6)

3.确定CUDA Runtime (对应下面都支持)

4. 看自己的驱动CUDA Driver Version

     搜索cmd(命令提示符)----nvidia-smi

5.最终确定适用CUDA 11.1-11.6

*******************************************************************************************

一、

     1.安装显卡驱动最新版本--nvidia官网

     2.打开命令窗口,输入nvidia-smi确定cuda driver版本(本人:11.6)

      3.打开Pytorch官网,确定CUDA Runtime 版本

      Previous PyTorch Versions | PyTorch,找到低于11.6版本的pytorch

二、

        在创建的虚拟环境中安装Pytorch(安装Pytorch, 安装3个包,直接从官网找到对应的conda版本/pip版本文件)

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

三种途径:①直接anaconda prompt

                      ②从pycharm终端(Terminal)直接输入

                      ③本地下载,然后利用pip install 路径地址安装

                         (将地址复制到迅雷里,这里的路径地址是指下载后文件所在地址)

三、验证pytorch是否成功

  ①激活对应的虚拟环境(conda activate 虚拟环境名)

  ②输入conda list,看有没有pytorch或torch

  ③输入python

   ④输入import torch-----回车    输入torch.cuda.is_available()-----如果返回True,说明成功(可以使用GPU版本)

  

********每天进步一丢丢,知识来源:b站:我是小土堆(仅供学习使用,用于自己对知识的总结)

              

  • 7
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值