环境配置问题一些注意的点(Pytorch+cudnn)
1、先安装好Pycharm以及Anaconda
我使用的是Python3.6版本,所以下载了与Python3.6对应的Anaconda,Pycharm下载的是社区版;
2、cudnn版本:首先查看自己电脑NVIDIA版本,从任务管理器中查看GPU型号,再从cudnn官网下载相关配件;
3、在notebook中配置合适的虚拟环境,命令为:conda create -n pytorch_gpu python=3.6
4、打开虚拟环境,在里面pip install所需的pytorch库的版本,检验其是否正确安装的命令为:
点击查看代码
import torch
print(torch.cuda.is_available())
5、第4步完成后,进入Pycharm中,导入创建好的虚拟环境。