环境配置问题一些注意的点(Pytorch+cudnn)

发布时间 2024-01-06 12:19:46作者: 首心

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())
若结果为True,则安装成功

5、第4步完成后,进入Pycharm中,导入创建好的虚拟环境。