pip安装cuda版torch

Posted 2023-04-27 不积跬步无以至千里

一开始安装torch的时候没有指定cuda版本,所以安装的是cpu版本,无法调用gpu

可以使用如下安装方法:

pip install torch -f https://download.pytorch.org/whl/cu118/torch_stable.html(应该可以用)

pip install  torch  --index-url https://download.pytorch.org/whl/nightly/cu118(亲测有效,但是nightly表示PyTorch 开发人员每天编译和生成的版本，包含了最新的代码和功能,不稳定)

本文来自博客园，作者：海_纳百川，转载请注明原文链接：https://www.cnblogs.com/chentiao/p/17357928.html，如有侵权联系删除

安装fastnlp

直接 pip install fastnlp

如果出现 torch的安装报错，可能与操作系统及 CUDA 的版本相关。直接上pytorch 的官网

https://pytorch.org/get-started/locally/

 

 

 用下面的那段文字，pip  或 pip3

最后再粗暴的直接    pip install fastnlp