Python神经网络环境搭建(anaconda,cuda,jupyter,tensorflow,pytorch,torchvision,pycharm)

1. 安装Anaconda

Anaconda清华源下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

在Anaconda Prompt创建虚拟环境:
conda create -n 虚拟环境名(自定义) python=版本号,具体代码:

conda create -n test2 python=3.7

2. 安装CUDA

确定GPU能安装什么版本的CUDA:
首先看自己电脑的显卡:打开任务管理器,点击性能
打开Anaconda的Anaconda Prompt输入命令:nvidia-smi
查看GPU能够支持的最高CUDA版本,我的最高支持到12.0
在这里插入图片描述
安装CUDA:
选择能够支持的版本下载cuda,下载地址:https://developer.nvidia.com/cuda-toolkit-archive
我选择的是11.6版本,下载完后就点击下一步安装即可。
安装完成后,打开cmd输入命令nvcc -V可以查看CUDA版本。
在这里插入图片描述
但是!如果显示的是:‘nvcc’ 不是内部或外部命令,也不是可运行的程序或批处理文件。
在这里插入图片描述
那首先去检查一下CUDA有没有安装成功。找到安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite。找到下面两个文件
在这里插入图片描述
以管理员身份打开cmd,通过cd命令转换到相应的文件位置,输入上图圈中的命令进行测试,若会显示result=pass,则说明安装无误,那么应该是在安装之后没有设置好系统环境变量,重新设置变量之后再进行测试。
添加环境变量:
在这里插入图片描述
在这里插入图片描述
添加到PATH中:
在这里插入图片描述
然后:打开cmd,输入:set PATH=c:
继续输入:echo %PATH%
关闭“cmd"后再次开启cmd,输入echo %PATH%
就会发现新添加的环境变量已经生效,输入nvcc -V
这样就可以正常看到CUDA的信息了:
在这里插入图片描述

3.安装pytorch

CUDA版本确定后,pytorch要依赖此CUDA版本,因为pytroch安装跟cuda对应的。
在这里插入图片描述
pytorch安装下载的地址:https://download.pytorch.org/whl/torch_stable.html

因为我安装的是11.6的CUDA,安装的python版本是3.10,所以对应安装pytorch:
在这里插入图片描述
其中cu代表cuda版本,cp代表python版本,torch代表要安装的pytorch版本。
我们这边是离线安装,会更加稳定更快。
首先要激活安装pytorch的python环境,之后到下载目录下进行安装。

D:\anaconda3\conda_env>pip install "torch-1.13.1+cu116-cp310-cp310-win_amd64.whl" -i https://pypi.tuna.tsinghua.edu.cn/simple/

在这里插入图片描述
安装成功后测试一下:

import torch
print(torch.__version__)  #查看pytorch的版本
print(torch.cuda.is_available()) #查看GPU能否使用,能则返回TRUE

在这里插入图片描述

4.安装torchvision

如果要使用计算机视觉,就要安装torchvision。这边依旧采用离线安装,先下载后安装。还是相同的网址:https://download.pytorch.org/whl/torch_stable.html
torch与torchvision也有对应关系,我是1.13.1的torch,3.10的python,所以我选择对应的
0.14.1的torchvision。
在这里插入图片描述
找到对应的下载链接:
在这里插入图片描述
首先要激活安装pytorch的python环境,之后到下载目录下进行安装。

pip install "torchvision-0.14.1+cu116-cp310-cp310-win_amd64.whl" -i https://pypi.tuna.tsinghua.edu.cn/simple/

在这里插入图片描述
安装完毕后可以查看一下安装列表conda list
在这里插入图片描述

4.安装jupyter

安装anaconda后会自动安装了jupyter notebook,但是这个jupyter是在base环境下的。为了让jupyter使用pytorch,就要在我创建的虚拟环境中下载。
anaconda prompt中激活我创建的ai_env环境,之后依次输入:

conda install ipython
conda install jupyter

打开jupyter后发现无法在Kernel中切换环境,再打开anaconda prompt,在base环境下输入命令conda install nb_conda,安装nb_conda,之后在jupyter中测试成功。

5.安装pycharm

直接在官网下载https://www.jetbrains.com/pycharm/

  • 3
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

你是个什么橙

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值