Mathematica中文论坛-非官方

 找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
热搜: 活动 交友 discuz
查看: 848|回复: 0

豆粑粑 ubuntu install torch

[复制链接]

442

主题

491

帖子

2411

积分

金牌会员

Rank: 6Rank: 6

积分
2411
发表于 2020-12-23 21:33:15 | 显示全部楼层 |阅读模式
需要安装torch,调用 gpu
创建一个环境


env_meat.yml
  1. name: th_meat
  2. dependencies:
  3. - matplotlib
  4. - tensorflow-gpu
复制代码
conda env create -f env_meat.yml

conda activate th_meat

  1. conda install -c pytorch torchvision cudatoolkit=10.1 pytorch
复制代码


回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|Archiver|手机版|小黑屋|Mathematica中文论坛-非官方 ( 辽ICP备16001491号-1

GMT+8, 2021-1-24 21:06 , Processed in 0.113658 second(s), 24 queries .

Powered by Discuz! X3.2

© 2001-2013 Comsenz Inc.

快速回复 返回顶部 返回列表