很多文章没介绍 CUDA是什么,上来直接安装,没有创建对应的环境。
so! 自己总结一下
转载请贴链接 https://blog.csdn.net/u013271656/article/details/105054615
1. Anaconda下创建torch环境
1.创建torch虚拟环境
>> conda create -n your_env_name python=X.X(3.7版本)
2. 激活虚拟环境
>> source activate your_env_name(虚拟环境名称)
2. 安装 torch
1. 进入 pytorch 官网, https://pytorch.org/
2. 选择自己所需版本,其中 CUDA 为安装GPU版本所需参数,可根据自身情况选择
3. 我选择CPU版本 CUDA==none
4. 复制命令,开始安装 Run this Command:
>> conda install pytorch torchvision cpuonly -c pytorch
3. 验证安装是否成功
check version:
1. 输入 python # 打开 python 开始编译
2. >> import torch
3. >> torch.__version__
# 成功输出版本号,安装完成,bingo
4. 安装spyder编译器
1. 查看所安装的库
>> conda list
2. 安装本环境下 python 版本所对应的 spydr
>> conda install spyder
3. 安装完成后输入 spyder ,并创建脚本
>> import torch
>> print(torch.__version__)
run
>> ^-^! 一切完成
ps: 安装了多个环境,如 tensorflow \ torch 等,每个环境对应的python版本可能存在差异
所以打开环境后, conda list #check一下各类库
作者:回收二手手机