栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Anaconda+Pytorch1.1+Cuda10.0+Cudnn7.6

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Anaconda+Pytorch1.1+Cuda10.0+Cudnn7.6

Anaconda
  • 下载地址:Anaconda
  • 创建虚拟环境:Conda虚拟环境管理
    • conda create -n torch_12 python=3.6
    • 上方的torch_12为虚拟环境名,可自定义
安装Cuda和Cudnn
  • Cuda下载地址:Cuda10.0
  • Cudnn下载地址(需要登录):Cudnn

注意:Cudnn的版本需要与Cuda对应

  • Cuda安装
    • 运行exe安装包
    • 自定义安装
    • 安装Cuda即可(驱动等其他东西一般windows已经安装了最新版)
  • 安装Cudnn
    • 解压zip压缩包,复制所有文件到Cuda安装目录下,如:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0
  • 验证安装:
    • Cuda: nvcc -V

    • Cudnn: nvidia-smi

安装pytorch1.2-Cuda10.0
  • 安装命令查询:pytorch安装命令查询

  • 进入虚拟环境:conda activate torch_12

  • 安装:conda install pytorch1.1.0 torchvision0.3.0 cudatoolkit=10.0 -c pytorch

  • 验证(输出true表示安装成功)

# 输入python命令进入python环境
import torch
import torchvision
print(torch.cuda.is_available())

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/354912.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号