CUDA下载:https://developer.nvidia.com/cuda-downloads
检查CUDA安装是否成功:
CMD输入nvcc -V
如果出现版本号等信息则说明CUDA安装成功
PyTorch安装注意:
官网的命令行复制的时候把=号旁边的空格去掉!!!
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
粘贴到cmd时=号旁边会有空格,Anaconda会显示没找到对应的包!!!
CUDA下载:https://developer.nvidia.com/cuda-downloads
检查CUDA安装是否成功:
CMD输入nvcc -V
如果出现版本号等信息则说明CUDA安装成功
PyTorch安装注意:
官网的命令行复制的时候把=号旁边的空格去掉!!!
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
粘贴到cmd时=号旁边会有空格,Anaconda会显示没找到对应的包!!!