Torch 1.10.1
- 下载并安装Cuda11.3
- 下载并安装Cuda11.3
CUDA Toolkit 11.3 Downloads | NVIDIA Developer
- 下载miniconda
Miniconda — Conda documentation
- 进入conda命令行 创建虚拟环境
conda create -npython=3.8 #创建虚拟环境 conda active #激活虚拟环境 conda deactive #退出虚拟环境
- 在虚拟环境中安装torch:
Previous PyTorch Versions | PyTorch
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
- Pycahrm配置
(1)、添加新的解释器
(2)、已经存在的环境
(3)、找到envs下的虚拟环境pyrorch1.10下的python.exe
OK!
- 测试
import torch print(torch.cuda.is_available()) print(torch.__version__) print(torch.version.cuda) print(torch.backends.cudnn.version())
在虚拟环境中安装jupyter notebook
conda active#进入虚拟环境 conda install nb_conda #安装jupyter jupyter notebook #进入



