本文只针对我在安装过程中遇到的问题,仅供参考。具体安装教程请参考pytorch官网(https://pytorch.org/get-started/locally/)。
第一次安装:因为电脑已经安装了CUDA,所以直接按照官网的指令安装,如下图蓝色框所示所示:
网速一定要好。显示安装成功之后,做测试。
为确保正确安装PyTorch,在这里,我们将构造一个随机初始化张量。如下图所示,可以正确运行。
但是,当检查PyTorch是否启用了GPU驱动程序和CUDA,是否可以通过PyTorch访问时,遇到了问题。
结果为False。
我猜测是CUDA版本不匹配导致的。开始安装PyTorch之前,我使用nvcc -V命令查看CUDA的版本,如下图所示:
根据上面的结果,我直接按照CUDA为10.0的版本安装,即执行命令conda install pytorch torchvision cudatoolkit=10.0 -c pytorch。
###输入命令:nvidia-smi 查看显卡版本以及CUDA版本。
结果显示错误:无法连接NVIDIA驱动:NVIDIA-SMI has failed because it couldnt communicate with the NVIDIA driver......
解决办法:参考链接:https://www.jianshu.com/p/3cedce05a481
从上图可以看到,CUDA的版本是10.1
###重新安装:conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
这次成功安装。
###重新安装过程中遇到的问题。
1、Solving environment: failed with initial frozen solve. Retrying with flexible solve....
参考链接:https://blog.csdn.net/Brookekitty/article/details/106226285