用conda安装pytorch和cudatoolkit

• Creating a conda Environment for Pytorch and Cuda - YouTube

• 安装conda并设置配置文件.condarc

• conda创建新python虚拟环境 >

conda create --name cuda_test python=3.8 -y conda activate cuda_test

• conda安装cudatookit，这个要和pytorch一起安装吗？ >

conda install cudatoolkit -c anaconda -y # 我理解cudatoolkit会自动和cuda版本匹配吧？

• conda安装pytorch相关 >

Start Locally | PyTorch
conda install pytorch-cuda=11.8 -c pytorch -c nvidia -y # 先单独安装pytorch-cuda，不要一起
conda install pytorch torchvision torchaudio -c pytorch -c nvidia -y
测试安装结果
python
import torch
torch.cuda.is_available()
a = torch.randn()
import numpy as np a = np.random.randn(5) b = torch.from_numpy(a) b.to("cuda") # 把gpu也看做一种软件可能更好理解，这样的话cuda相当于和另外一个软件进行交互操作