pytorch是在conda环境中还是在系统中使用cudatoolkit？

izkcnapc 于 2022-11-09 发布在其他

关注(0)|答案(1)|浏览(152)

我在conda环境中使用pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101安装了pytorch和torchvision，据我所知，这意味着pytorch库是使用cuda10.1编译的。

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

我假设这意味着我的系统中的cudatoolkit是cuda9.1，但是如果我接着在我的conda环境中用conda install -c anaconda cudatoolkit=10.1安装一个不同版本的cudatoolkit，pytorch会使用哪个cudatoolkit？
我使用pip安装pytorch，因为这是在原来的回购我计划使用的指示。

pytorch

来源：https://stackoverflow.com/questions/70479396/does-pytorch-use-the-cudatoolkit-in-the-conda-environment-or-the-system