欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

pytorch安装

程序员文章站 2022-05-31 18:16:55
...

1.确定cuda版本:

 cat /usr/local/cuda/version.txt

2.在pytorch官网确定代码:

  conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

3.若在服务器上conda demand not found:

export PATH=~/anaconda3/bin:$PATH

上一篇: nextTick的使用

下一篇: SVN安装