查看搭建的cuda环境

cuda 11.7   torch 2.0.1+cu117     cudnn 8500    Python 3.9.16

第一步:cmd打开命令窗口,输入conda-env list 查看有几个conda环境

第二步:选择一个环境进入conda,输入conda activate pytorch

第三步:进入python环境,输入python

第四部 :在Python中,查看torch、cuda和cudnn的版本

1. 查看torch版本

>>> import torch
>>> print(torch.__version__)

2.查看cuda版本

>>> print(torch.version.cuda)

3.查看cudnn版本

>>> print(torch.backends.cudnn.version())

 

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在Linux环境搭建CUDA的步骤如下: 1. 首先,需要安装CUDA和CUDNN。根据需要安装的CUDA版本,参考适当的安装教程以及官方文档。可以从NVIDIA官方网站下载CUDA安装包,并按照说明进行安装。 2. 安装完毕后,可以使用命令nvcc -V和nvidia-smi来验证CUDA的版本。在某些情况下,通过这些命令显示的CUDA版本可能与实际运行时使用的版本不同。可以使用import torch和torch.utils.cpp_extension.CUDA_HOME命令来查看PyTorch实际使用的CUDA版本。 3. 如果想切换CUDA版本,可以在.bashrc文件中进行修改。打开.bashrc文件,添加以下代码: ``` export PATH="/usr/local/cuda-<version>/bin/:$PATH" export LD_LIBRARY_PATH="/usr/local/cuda-<version>/lib64:$LD_LIBRARY_PATH" export CUDA_HOME=/usr/local/cuda-<version> ``` 其中,<version>替换为你想要使用的CUDA版本号。保存并关闭.bashrc文件。 4. 为了使修改后的.bashrc文件生效,需要在终端中运行以下命令: ``` source ~/.bashrc ``` 这样,修改后的CUDA环境就会生效。 5. 最后,可以使用nvcc -V和nvidia-smi命令来验证CUDA版本是否已成功切换。 需要注意的是,上述步骤仅适用于在用户自己的环境中切换CUDA版本。如果在共享服务器上使用,可能需要在/etc/profile文件中进行修改,以改变默认的CUDA版本。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* [【转载】Linux安装CUDA 和 cuDNN,以及CUDA个人环境和系统环境的切换](https://blog.csdn.net/a_piece_of_ppx/article/details/129477191)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *3* [服务器Linux环境配置cuda(非管理员)](https://blog.csdn.net/ffftttxxx/article/details/125910239)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值