pytorch_GPU安装

什么是CUDA?
Pytorch安装(GPU版本)
CUDA与cudnn的关系

联想拯救者R9000P 安装torch_GPU 只需要首先安装anconda,创建环境,在该环境下,直接使用一下命令即可:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 
或者:
pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 torchaudio==2.0.2 -f https://download.pytorch.org/whl/torch_stable.html

换镜像源

### 设置清华源镜像
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
## 设置搜索时显示通道地址
 conda config --set show_channel_urls yes
### 设置pytorch镜像
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

恢复默认镜像源

conda config --remove-key channels

使用源安装一些包的命令
pip install -i 镜像源 设置信任镜像源 包名

pip install -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com d2l

验证

import torch
  
torch.cuda.is_available()
True
  
torch.cuda.current_device()
0
  
torch.cuda.device(0)
<torch.cuda.device object at 0x0000023034332438>
  
torch.cuda.device_count()
1
  
torch.cuda.get_device_name(0)
'GeForce GTX 1660'

查看版本

import torch
torch.__version__
import torchvision
torchvision.__version__

问题

1、(不是镜像问题,是命令问题!)Anaconda安装Pytorch报错:PackagesNotFoundError

Anaconda安装Pytorch报错:PackagesNotFoundError 的解决办法

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要在Ubuntu上安装PyTorch GPU版本,可以按照以下步骤进行操作: 1. 首先,确保你已经安装了NVIDIA显卡驱动程序。可以通过运行以下命令来检查是否安装了正确的驱动程序版本: ``` nvidia-smi ``` 这将显示你当前安装的NVIDIA驱动程序的版本信息。 2. 接下来,你需要安装CUDA工具包。可以通过以下步骤进行安装: a. 根据你的显卡型号和操作系统版本,在NVIDIA官方网站下载对应版本的CUDA工具包。 b. 运行下载的安装程序,并按照提示进行安装。注意选择合适的安装选项,以确保安装了CUDA驱动程序和CUDA工具。 c. 安装完成后,将CUDA的bin目录添加到系统的PATH环境变量中,这样你就可以在任意位置运行CUDA相关的命令。 3. 然后,你需要安装cuDNN库。cuDNN是一个针对深度学习任务进行加速的GPU库。可以按照以下步骤进行安装: a. 前往NVIDIA开发者网站,注册一个账号并下载合适版本的cuDNN库。 b. 解压下载的cuDNN库文件,并将其中的bin、lib和include文件夹分别复制到对应的CUDA安装目录下。 4. 现在,你可以使用Anaconda来安装PyTorch。可以按照以下步骤进行安装: a. 打开终端,运行以下命令创建一个新的conda环境: ``` conda create -n pytorch_gpu ``` b. 激活该环境: ``` conda activate pytorch_gpu ``` c. 运行以下命令安装PyTorch和相关的依赖库: ``` conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch ``` 这将安装PyTorch GPU版本以及其他必要的依赖库。 5. 完成以上步骤后,你就成功在Ubuntu上安装PyTorch GPU版本。 请注意,上述步骤中的CUDA版本和cuDNN版本可能需要根据你的具体显卡和操作系统进行调整。建议在安装之前查阅PyTorch官方文档和NVIDIA官方网站获取更详细的信息和指导。还可以参考引用和引用[2]中提供的安装命令和参考资料。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值