1 配置问题
(1).针对于电脑中配备有GPU,且有深度学习需求,搭建一个可用无污染的深度学习环境。
(2).按照Anaconda3 + CUDA10.0 + CuDNN + Pytorch1.2 + Pycharm配置。
(3).解决配置Pytorch中无法使用torchvision库的问题。
2 搭建过程
(1).安装Anaconda。按照自己电脑的需求下载,下载完成后,解压安装包,按照指示一步一步完成即可,检验安装是否成功,并创建一个虚拟环境用于之后的配置。
# 创建虚拟环境:(在cmd终端输入) conda create -n Pytorch_GPU python=3.7 |
(2).安装CUDA10.0和cuDNN(必须与自己的Pytorch==1.2.0版本对应起来)
activate Pytorch_GPU
2).下载安装Pytorch。打开pytorch的官网,地址:https://pytorch.org/。
也可以去下载轮子(https://download.pytorch.org/whl/torch_stable.html)
(4)Pycharm和Anaconda连接。
1.要注意各个软件版本之间的匹配问题。
2.最新版本的CUDA不再需要配置Cudnn进行加速了。
3.针对torchvision导入模块缺失问题,更改__init__.py文件的导入模块名,就可以使用了。