Mathematica中文论坛-非官方

 找回密码
 立即注册

QQ登录

只需一步,快速开始

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

豆粑粑 ubuntu install torch

[复制链接]

525

主题

594

帖子

2980

积分

金牌会员

Rank: 6Rank: 6

积分
2980
发表于 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, 2024-3-29 19:40 , Processed in 0.108938 second(s), 24 queries .

Powered by Discuz! X3.2

© 2001-2013 Comsenz Inc.

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