豆粑粑 ubuntu install torch

meatball1982

需要安装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

复制代码