Linux/Ubuntu下查看cuda、python、torch、torchvision与torchtext安装
我的torch是1.5.0,对应安装的torchtext是0.6.0。根据torch版本找到对应的torchtext版本,进行安装。必须先进入到python中,此时其实也能查看python版本。查看torchvision。利用官网给出的代码安装。
·
1.查看cuda版本
ubuntu下查看cudnn版本+cuda版本-CSDN博客
nvcc -V
2.查看python版本
python -V
3. 查看torch、torchvision版本
必须先进入到python中,此时其实也能查看python版本
python
查看torch
import torch
torch.__version__
查看torchvision
import torchvision
torchvision.__version__
4.退出python环境
exit()
5.torchtext安装
根据torch版本找到对应的torchtext版本,进行安装
我的torch是1.5.0,对应安装的torchtext是0.6.0
利用官网给出的代码安装
pip install torchtext==0.6.0
欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
已为社区贡献2条内容
所有评论(0)