因为之前已经安装过tensorflow-gpu
所以我的cndu与cudnn与annaconda都已经安装好
接下来安装pytorch也是基于conda的虚拟环境
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
//如果在配置的过程中发生了什么问题,可以先退出虚拟环境然后移除该虚拟环境,再重新配置
conda remove -n pytorch --all
//主要参考帖子:
https://blog.csdn.net/hktxt/article/details/82111203
//但是我的pytorch安装命令来自官网:
https://pytorch.org/get-started/previous-versions/
# CUDA 9.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
// 因为已经把源换成清华,因此这个命令在安装pytorch时还是从官网下载,很慢,参考帖子把-c去掉:
https://blog.csdn.net/jiyangsb/article/details/82430794
问题:
好像在虚拟环境中又重新给我安装了cuda与cudnn