1.下载Anaconda
集成了很多科学分析包的环境集成软件
2. 下载pytorch
主要支持英伟达显卡,显卡起到训练加速配置
需要检查显卡驱动是否正确安装
当资源管理器中能看到显卡型号时代表显卡驱动安装成功
打开Anaconda Prompt输入以下指令
安装包环境
conda create -n pytorch python=3.6
进入到pytorch下
conda activate pytorch
检查包
pip list
查询是否安装成功
我之前返回false的原因是torch版本和cuda不对应,通过安装包安装1.3.0版本后成功解决问题。
3. pycharm导入环境
测试torch可用
4. jupyter导入pytorch环境
conda activate pytorch
在激活pytorch环境下执行指令:conda install nb_conda或pip install --upgrade --user nbconvert