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版本,进行安装

官网 torchtext · PyPI

 我的torch是1.5.0,对应安装的torchtext是0.6.0

利用官网给出的代码安装

pip install torchtext==0.6.0

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐