← 返回首页
pytorch与torch的关系
发表时间:2024-02-20 02:00:23
pytorch与torch的关系

Pytorch是Torch在Python上的衍生。因为Torch是一个使用Lua语言的神经网络库,Torch很好用,但是Lua不是特别流行,所有开发团队将Lua的Torch移植到了更流行的语言Python上。

1.torch与pytorch的区别

总结来说,Torch是一个以Lua语言为主的科学计算框架,而PyTorch是Torch的Python接口,提供了动态计算图等更高级的功能。

2.PyTorch GPU版本的安装

1).pytorch和torch包有何区别? 没有区别,在pip管理的包名叫torch,在conda管理的包名叫pytorch,注意安装时输入包的正确名称。 2).如何安装正确的CUDA版本? 如果对GPU对应安装版本有疑惑,可以先用nvidia-smi查看目前机器GPU使用的CUDA版本。只要安装的包比当前机器版本低都没有问题。

3).如何正确升级conda版本以获取最新版本?

以linux为例,首先要赋予正确的写权限

sudo chown -R [username] /path/miniconda3

再升级conda版本

conda update -n base -c defaults conda --repodata-fn=repodata.json