当前位置:   article > 正文

torch安装_torch库安装

torch库安装

1、下载python和cuda,确定python版本,cuda版本

Python: 3.8

Cuda:10.1

对应torch:

https://blog.csdn.net/FL1768317420/article/details/134769203?ops_request_misc=&request_id=&biz_id=102&utm_term=torch1.13.0%E5%AF%B9%E5%BA%94cuda&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-1-134769203.142^v100^control&spm=1018.2226.3001.4187

2、根据对应版本找到torch、torchvision对应版本(注意不要错下载成cpu的);

     https://download.pytorch.org/whl/torch_stable.html

(这种是cpu的)

     找到torch_scatter:

https://pytorch-geometric.com/whl/

3、点击下载.whl文件(记好文件所在位置);

4、在黑框中先转到相应位置,再进行pip install 或者conda install;

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/爱喝兽奶帝天荒/article/detail/880396
推荐阅读
相关标签
  

闽ICP备14008679号