关于Compound Word Transformer论文代码的环境配置

最新推荐文章于 2024-03-18 14:16:27 发布

YoRoll_町

最新推荐文章于 2024-03-18 14:16:27 发布

阅读量556

点赞数

分类专栏：深度学习文章标签： transformer 深度学习 python

本文链接：https://blog.csdn.net/Gragon_sgirlfriend/article/details/125767660

版权

深度学习专栏收录该内容

4 篇文章 0 订阅

订阅专栏

代码作者放在这里
只需要配置好环境就可以将代码运行起来了，代码对于环境版本有严格的要求。
1、先在服务器上配置版本号10.1的cuda（不知道是只更改虚拟环境的版本还是系统和虚拟环境的版本都需要更新，所以我全部都进行了更改）
因为后续的pytorch-fast-transformer的安装，需要10.1以上的cuda。
在这里借鉴到的
首先，输入nvcc -V查看当前版本，如果是10.1的版本，我们只需要在虚拟环境安装相同版本的cuda和pytorch，如果不是我们最好对系统版本进行更新，方法如下。

nvidia-smi

检查是否支持安装10.1的cuda。
参考以下博客进行cuda的版本更新。服务器中安装多版本的cuda

2、接着，进行虚拟环境cuda的安装，我参考以下链接。
conda虚拟环境中安装cuda和cudnn，再也不用头疼版本号的问题了
3、最后我们输入以下命令，分别安装pytorch、pytorch-fast-transformer、miditoolkit（注意版本号要一致，不一致会出问题）

pip install torch===1.4.0+cu100 torchvision===0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html
pip install pytorch-fast-transformer==0.3.0
pip install miditoolkit==0.1.14