anaconda安装pytorch-gpu版本

1.安装cuda环境

1.1 更新显卡驱动(非必需,只要步骤1.2中的CUDA版本支持11.6及以上就可以)

打开NVIDIA控制面板,可以看到显卡型号:GeForce RTX 2060,显卡驱动版本:512.78

进入NVIDIA官网检索最新驱动版本并更新(我偷懒没更新):https://www.nvidia.cn/Download/index.aspx?lang=cn

1.2 查看显卡对应的CUDA版本

nvidia控制面板->帮助->系统信息->组件,可以看到CUDA版本是11.6.134

1.3 下载并安装对应版本的CUDA,推荐本地安装(local)

https://developer.nvidia.com/cuda-toolkit-archive

1.4 下载并安装对应版本的cuDNN,需要注册一个账号

https://developer.nvidia.com/rdp/cudnn-archive

2. 下载并安装anaconda

https://www.anaconda.com/

2.1 在conda虚拟环境中安装python

打开Anaconda Powershell Prompt

输入命令conda create -n env_name python=x.x

conda create -n pytorch python=3.7

……

3. 安装pytorch

方法一:进入官网找到对应版本的安装命令,如果网速很慢的话会需要换源

https://pytorch.org/get-started/locally/

方法二:离线下载安装包到本地,再进行离线安装

https://download.pytorch.org/whl/torch_stable.html

……

4. 安装pycharm专业版,使用学校邮箱或学信网在线报告进行认证免费使用,一年认证一次。配置pycharm项目的解释器。

  • 0
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值