pycharm 安装 pytorch 验证cuda

进入Pytorch官网
选择自己的配置,官网会提示两行命令,照着做就行了

pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-linux_x86_64.whl

验证cuda

import torch

print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.get_device_name())
print(torch.cuda.current_device())
  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要配置PyTorch环境并在PyCharm中使用PyTorch,可以按照以下步骤进行操作: 1. 首先,需要安装Anaconda和PyCharm。您可以从官方网站下载并按照说明进行安装。 2. 接下来,打开PyCharm,创建一个新的PyTorch环境。点击菜单栏中的"File",然后选择"Settings"。在弹出的窗口中,点击"Project Interpreter"。 3. 在"Project Interpreter"页面的右上角,点击齿轮图标,选择"Add"。然后选择"Conda Environment",并输入环境的名称。 4. 在选择Python版本时,选择与您的PyTorch版本兼容的Python版本。如果您不知道自己的CUDA版本号,可以通过控制面板中的NVIDIA控制面板找到。 5. 创建环境后,可以在"Terminal"终端中进行PyTorch安装。在PyCharm的底部工具栏中,点击"Terminal"图标,打开终端。 6. 在终端中,输入以下命令来安装PyTorch: ``` conda install pytorch torchvision cpuonly -c pytorch ``` 7. 安装完成后,您可以在PyCharm验证是否成功安装PyTorch。在终端中,输入以下命令: ``` python >>> import torch ``` 如果没有报错,则说明PyTorch已成功安装。您可以继续输入以下代码来检查是否可以使用GPU: ``` >>> torch.cuda.is_available() ``` 如果输出为True,则表示您可以使用GPU加速了。 8. 现在您已经成功配置了PyTorch环境并可以在PyCharm中使用PyTorch了。 总结: 1. 安装Anaconda和PyCharm。 2. 在PyCharm中创建一个新的PyTorch环境。 3. 在终端中输入命令安装PyTorch。 4. 验证安装是否成功。 请注意,根据您的具体情况和操作系统,步骤可能会有所不同,以上步骤仅供参考。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [Pycharm中切换pytorch的环境和配置](https://download.csdn.net/download/weixin_38661100/13751693)[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: 33.333333333333336%"] - *2* [pycharm配置pytorch](https://blog.csdn.net/qq_41511783/article/details/124454132)[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: 33.333333333333336%"] - *3* [pytorch安装、环境搭建及在pycharm中的设置](https://blog.csdn.net/laozaoxiaowanzi/article/details/104804164)[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: 33.333333333333336%"] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值