虚拟环境与conda以及GPU使用

1. 在虚拟环境中安装GPU版本torch

1.1 需要在win系统下安装匹配版本的CUDA toolkit(toolkit就是指工具包)和 cuDNN(用于配置深度学习使用)

需要查看显卡驱动最高支持的版本,两种方法:
1.桌面右键NVIDIA控制面板
2.cmd nvidia-smi
如图所示CUDA版本最高到12.0,向下兼容。
(但是要注意torch版本同样要与CUDA和cudnn兼容,目前2023.7torch兼容到了CUDA11.8版本)

在这里插入图片描述

在这里插入图片描述

CUDA

官网链接:CUDA Toolkit Archive
这里选择win11,11.8版本
在这里插入图片描述
在这里插入图片描述
检验CUDA是否安装成功
cmd nvcc -V
在这里插入图片描述

cudnn

官网链接:cuDNN Archive
这里选择与CUDA 11.8 匹配的最新版本
在这里插入图片描述
CUDA和cudnn详细安装步骤见CUDA与cuDNN安装教程(超详细)

1.2 在虚拟环境中安装GPU的torch,torchvision,torchaudio

方法一:安装包下载到本地,离线安装到虚拟环境中
方法二:(抄来的,未测试)

pip3 install torch==2.0.0+cu118 torchvision==0.15.0+cu118 torchaudio==2.0.0 -f https://download.pytorch.org/whl/torch_stable.html

只介绍方法一

step1:

首先创建一个虚拟环境,创建时选择对应的python版本
关于conda相关指令,见:Conda常用命令记录与.condarc

step2:

安装包地址:pytorch.org
torch示例:whl与tar,gz区别?
在这里插入图片描述

cu118:CUDA 11.8
torch-2.0.1:torch版本
cp311:python3.11
win:windows

下载好的三个文件如下:(下载的是2.0.0版本的torch)
在这里插入图片描述

step3:

进入到虚拟环境
cd 到文件所在目录
在这里插入图片描述
我的这三个文件所在目录是D:\Download_mydata\Edge_download
如下图一个一个pip install 就可以
在这里插入图片描述

测试:

import torch

if torch.cuda.is_available():
    device = torch.device("cuda")
    print("CUDA is available.")
    print("Device:", torch.cuda.get_device_name())
else:
    device = torch.device("cpu")
    print("CUDA is not available. Using CPU.")

在这里插入图片描述

step4:

给新环境安装jupyter notebook和ipykernel,这里jupyter只是一个交互界面,ipykernel是内核控制(大概是这个意思,详见IPython与Jupyter Notebook

conda install jupyter
conda install ipykernel

在环境里执行下面的代码来手动添加kernel进jupyter notebook

python -m ipykernel install --user --name [虚拟环境名] --display-name "kernel命名"

python -m ipykernel install --user --name py3.11 --display-name "py311kernel"

之后再jupyter里面就有了
在这里插入图片描述

step5: 可有可无

新环境下创建Jupyter Notebook的桌面快捷方式
就像这样,快速启动该环境下的jupyter notebook

观察之前的一个已经创建好快捷方式的环境”env_ML2“和base环境快捷方式的目标
在这里插入图片描述
复制到记事本对比
在这里插入图片描述
区别在于后三个,分别指向了虚拟环境文件夹、文件夹中的python、文件夹中的jupyter notebook
新建一个快捷方式,将目标改为如下:
注意没有回车,只有空格
在这里插入图片描述
全部换成D:\Code\Anaconda_files\py3.11\Scripts\jupyter-notebook.exe达到同样效果,不理解?

但是不知道step5意义是什么?
在这里插入图片描述

补充

在更新windows版本后,出现了cuda无法使用的问题,输入以下代码无法调用gpu

import torch

if torch.cuda.is_available():
    device = torch.device("cuda")
    print("CUDA is available.")
    print("Device:", torch.cuda.get_device_name())
else:
    device = torch.device("cpu")
    print("CUDA is not available. Using CPU.")

解决方法:
手动卸载torch-gpu三件套,重新安装。
1.从D:\Code\Anaconda_files\py3.11\Lib\site-packages中找到三个文件,连带包含dist-info的文件夹全部删除。(这里dist-info好像在虚拟环境安装包的时候都会有)

Python package dist-info 是 Python 包管理系统中一种用于存储元数据的目录结构。它在 Python 3.3 及以后的版本中作为安装 Python 包的一种标准方式被引入,用于替代旧的 .egg-info 目录。
dist-info 目录通常位于 Python 包的安装目录下,并包含一个名为 METADATA 的文件和一个名为 RECORD 的可选文件。其中,METADATA 文件包含了与该 Python 包相关的元数据信息,例如包名、版本号、作者、许可证等等。RECORD 文件记录了该包中每个文件的路径和校验和,以便在卸载或升级包时进行文件的清理或者更新。
Python 包的安装方式可以分为两种:系统范围安装和虚拟环境安装。在系统范围安装中,Python 包将会被安装到 Python 解释器的全局环境中,并且 dist-info 目录会被安装在 Python 的 site-packages 目录下。而在虚拟环境中,Python 包会被安装到虚拟环境对应的目录中,并且 dist-info 目录也会相应地被安装在虚拟环境的 site-packages 目录下。
总之,dist-info 目录是 Python 包管理系统中的一个标准化的目录结构,用于存储 Python 包的元数据信息和记录文件清单,它是 Python 包管理的一个重要组成部分。

在这里插入图片描述
2.重新上面的操作,进行安装。

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
### 回答1: 您可以参考以下步骤在Conda虚拟环境中安装TensorFlow GPU:1.首先,在您的Deepin系统中安装Conda。2.然后,使用Conda创建一个虚拟环境。3.接着,在虚拟环境中安装TensorFlow GPU。4.最后,激活您的虚拟环境,并开始使用TensorFlow GPU。 ### 回答2: 要在conda虚拟环境中安装tfGPU,需要按照以下步骤进行操作: 1. 首先,确保在你的计算机上已经安装了conda。如果没有,请先下载并安装最新的conda发行版。 2. 打开终端或命令提示符窗口,在命令行中输入以下命令,创建一个新的conda虚拟环境: ``` conda create -n myenv ``` 其中,"myenv"是你给虚拟环境起的名称,可以根据自己的喜好进行更改。 3. 激活你的虚拟环境。在命令行中输入以下命令: ``` conda activate myenv ``` 4. 接下来,你需要安装NVIDIA GPU驱动程序和CUDA和cuDNN库。你可以根据你的显卡和CUDA版本到NVIDIA官方网站下载并安装对应版本的驱动程序和CUDA和cuDNN库。 5. 安装TensorFlow GPU版本。在命令行中输入以下命令: ``` conda install tensorflow-gpu ``` 这将会自动下载和安装最新版本的TensorFlow GPU。请注意,这可能会花费一些时间来解决依赖关系和下载安装所需的文件。 6. 安装其他必要的依赖项。根据你的项目需求,你可能需要安装其他一些依赖项,例如Keras等。你可以使用conda包管理器来安装这些依赖项。 7. 确认安装成功。在命令行中输入以下命令,启动Python解释器: ``` python ``` 然后,导入TensorFlow库并验证是否成功: ```python import tensorflow as tf print(tf.__version__) print(tf.test.is_gpu_available()) ``` 如果输出显示了你安装的TensorFlow版本,并且指示GPU可用,则说明成功安装了tfGPU。 请注意,安装tfGPU需要相应版本的NVIDIA GPU驱动程序、CUDA和cuDNN库,以及与之兼容的TensorFlow版本。因此,请确保你按照官方文档中的要求进行安装,并根据你的硬件和软件环境进行相应的调整。 ### 回答3: 首先,为了在conda虚拟环境中安装tfGPU,我们需要确保已经在deepin操作系统上成功安装了Anaconda和CUDA。 1. 首先,在deepin操作系统上下载并安装Anaconda。可以在Anaconda官方网站上下载合适的版本,并按照官方指南进行安装。 2. 安装完成后,打开终端,并创建一个新的conda虚拟环境。可以使用以下命令: conda create -n myenv python=3.8 "myenv"是虚拟环境的名称,你可以自己设置。 3. 激活虚拟环境conda activate myenv 4. 接下来,安装CUDA。可以根据你的显卡型号和CUDA版本到NVIDIA官网上下载对应版本的CUDA,并按照官方指南进行安装。 5. 安装完成后,我们可以使用以下命令安装tensorflow-gpuconda install tensorflow-gpu 以上步骤将在你的conda虚拟环境中安装tfGPU。请注意,这些步骤需要一些时间和网络资源,所以请确保你的网络连接稳定。 安装完成后,你可以在python脚本或jupyter notebook中,导入tensorflow并开始使用tfGPU进行深度学习任务。确保在代码中使用`import tensorflow as tf`来导入tensorflow库,以便正确使用tfGPU加速功能。 希望以上信息能帮助你在deepin操作系统上成功安装tfGPU并进行深度学习任务。如果遇到任何问题,请参考conda和CUDA的官方文档或向相关社区求助。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值