conda一键安装pytorch-cuda环境的方法

采用cudatoolkit10.2更加兼容多数框架, 比如笔者正在使用的cupy。在windows下的user目录中的修改.condarc文件为。检索其他cudatoolkit版本的conda命令。3. 安装cupy-cuda。2. conda安装命令。1. 配置conda源。

maxOxs

240人浏览 · 2024-06-15 16:20:17

maxOxs · 2024-06-15 16:20:17 发布

1. 配置conda源

在windows下的user目录中的修改.condarc文件为

show_channel_urls: true
ssl_verify: true
channels:
- http://conda.anaconda.org/gurobi
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- conda-forge
- defaults
```

2. conda安装命令

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

采用cudatoolkit10.2更加兼容多数框架, 比如笔者正在使用的cupy

检索其他cudatoolkit版本的conda命令

conda search cudatoolkit

3. 安装cupy-cuda

pip install cupy-cuda102

Triton中文社区

欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

torch.device(‘cuda:0‘ if cuda else ‘cpu‘)

# 这个地方的cuda:0 实际上并不是0号GPU，他取决于CUDA_VISIBLE_DEVICES# 然后逻辑GPU和物理GPU有一个对应关系# 如果CUDA_VISIBLE_DEVICES为2,1,3# 那么CUDA:0就是2号GPU， CUDA:1 就是1号GPU CUDA:3 就是3号GPUreturn torch.device('cuda:0' if cuda else 'cpu')..