一、问题描述与原因
按照网上常见的安装pytorch的教程都是直接进入Pytorch官网寻找相应版本的pytorch安装命令代码,同时会安装cuda比如:
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
此时nvcc -v会出现错误
-bash: nvcc: command not found
这是因为使用conda install安装的cudatoolkit与NVIDIA打包的CUDA工具包不一样造成的。
二、解决方案
可以安装hcc版本的cuda,具体安装指令可以在该链接下寻找:Cudatoolkit :: Anaconda.org
相应conda指令比如安装cuda=10.2版本为如下:
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch -c hcc