conda换中科大源安装pytorch
时间: 2023-07-18 13:20:15 浏览: 127
您可以通过以下步骤在conda中使用中科大源安装pytorch:
1. 打开anaconda prompt或者终端,输入以下命令来添加中科大的conda源:
`conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/`
`conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/`
2. 更新conda:
`conda update conda`
3. 创建一个新的conda环境并安装pytorch:
`conda create --name myenv pytorch`
这将创建一个名为“myenv”的conda环境,并安装最新版本的pytorch。
如果您需要特定版本的pytorch,则可以使用以下命令:
`conda create --name myenv pytorch=1.7.1 torchvision cudatoolkit=10.1 -c pytorch -c https://mirrors.ustc.edu.cn/anaconda/pkgs/free -c https://mirrors.ustc.edu.cn/anaconda/pkgs/main`
这将创建一个名为“myenv”的conda环境,并安装pytorch 1.7.1、torchvision和cudatoolkit 10.1。
4. 激活环境并验证pytorch是否成功安装:
`conda activate myenv`
`python -c "import torch; print(torch.__version__)"`
如果输出pytorch的版本号,则表示已经成功安装。
注意:在使用中科大的conda源时,可能会因为网络问题导致下载速度较慢或者下载失败。如果遇到这种情况,可以尝试多次运行命令或者使用其他可靠的conda源。
阅读全文