Mathematica中文论坛-非官方

标题: 豆粑粑 ubuntu install torch [打印本页]

---

作者: meatball1982    时间: 2020-12-23 21:33
标题: 豆粑粑 ubuntu install torch
需要安装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

复制代码

---

欢迎光临 Mathematica中文论坛-非官方 (http://ilovemathematica.com/)

Powered by Discuz! X3.2