美文网首页
[安装三]linux-anaconda下安装pytorch

[安装三]linux-anaconda下安装pytorch

作者: 原来是酱紫呀 | 来源:发表于2018-11-28 00:03 被阅读0次

20181127 qzd


1. 添加清华源

然后使用conda进行安装(注:0.3.1 及以后)

### for 0.4.0 and later
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
(注: 添加的镜像源可以在 ~/home/你的系统名称/.condarc 文件中查看)

2. CPU版本

# for CPU only packages
conda install pytorch-cpu torchvision-cpu

3. GPU版本

(你的环境)>> pip install torch torchvision

4. 卸载pytorch

pip uninstall torch

相关文章

网友评论

      本文标题:[安装三]linux-anaconda下安装pytorch

      本文链接:https://www.haomeiwen.com/subject/fxeoqqtx.html